Home
last modified time | relevance | path

Searched refs:smaller_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py119 def _itGen(self, smaller_shape, larger_shape): argument
120 up_sample = (smaller_shape, larger_shape)
121 down_sample = (larger_shape, smaller_shape)
161 smaller_shape = [batch_size, 2, 3, channel_count]
163 for in_shape, out_shape, _, _ in self._itGen(smaller_shape, larger_shape):
192 smaller_shape = [batch_size, 2, 3, channel_count]
195 self._itGen(smaller_shape, larger_shape):
263 smaller_shape = [batch_size, 4, 6, channel_count]
265 for params in self._itGen(smaller_shape, larger_shape):
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc779 const Shape& smaller_shape, const Shape& larger_shape, in InferInDimBroadcastShape() argument
781 if (broadcast_dimensions.empty() && !ShapeUtil::IsScalar(smaller_shape)) { in InferInDimBroadcastShape()
786 ShapeUtil::HumanString(smaller_shape), in InferInDimBroadcastShape()
788 } else if (broadcast_dimensions.size() != smaller_shape.rank()) { in InferInDimBroadcastShape()
794 smaller_shape.rank(), broadcast_dimensions.size()); in InferInDimBroadcastShape()
837 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()
839 for (int i = 0; i < smaller_shape.dimensions_size(); ++i) { in InferInDimBroadcastShape()
852 int64 small_dimension_size = smaller_shape.dimensions(i); in InferInDimBroadcastShape()
854 bool small_is_dynamic = smaller_shape.is_dynamic_dimension(i); in InferInDimBroadcastShape()
865 ShapeUtil::HumanString(smaller_shape), in InferInDimBroadcastShape()
[all …]
Dshape_inference.h379 const Shape& smaller_shape, const Shape& larger_shape,
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_test_util.py1148 smaller_shape = batch_shape + [m, n - 1]
1159 smaller_shape, mean=0.0, stddev=stddev_mat, dtype=dtype, seed=seed)