Home
last modified time | relevance | path

Searched defs:compute_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_grad_filter_ops.cc911 TensorShape compute_shape = ShapeFromFormat( in operator ()() local
932 TensorShape compute_shape = ShapeFromFormat( in operator ()() local
H A Dconv_grad_input_ops.cc312 TensorShape compute_shape = ShapeFromFormat( in operator ()() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.cpp1230 void TensorIteratorBase::compute_shape(const TensorIteratorConfig& config) { in compute_shape() function in at::TensorIteratorBase