Searched refs:smaller_shape (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
D | image_grad_test_base.py | 119 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/ |
D | shape_inference.cc | 779 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 …]
|
D | shape_inference.h | 379 const Shape& smaller_shape, const Shape& larger_shape,
|
/external/tensorflow/tensorflow/python/ops/linalg/ |
D | linear_operator_test_util.py | 1148 smaller_shape = batch_shape + [m, n - 1] 1159 smaller_shape, mean=0.0, stddev=stddev_mat, dtype=dtype, seed=seed)
|