Home
last modified time | relevance | path

Searched refs:larger_shape (Results 1 – 3 of 3) 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)
122 pass_through = (larger_shape, larger_shape)
162 larger_shape = [batch_size, 4, 6, channel_count]
163 for in_shape, out_shape, _, _ in self._itGen(smaller_shape, larger_shape):
193 larger_shape = [batch_size, 5, 6, channel_count]
195 self._itGen(smaller_shape, larger_shape):
264 larger_shape = [batch_size, 8, 16, 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
787 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()
835 Shape output_shape(larger_shape); in InferInDimBroadcastShape()
837 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()
846 if (dimension_to_match >= larger_shape.dimensions_size()) { in InferInDimBroadcastShape()
850 dimension_to_match, larger_shape.dimensions_size()); in InferInDimBroadcastShape()
853 int64 large_dimension_size = larger_shape.dimensions(dimension_to_match); in InferInDimBroadcastShape()
856 larger_shape.is_dynamic_dimension(dimension_to_match); in InferInDimBroadcastShape()
866 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()
877 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()
[all …]
Dshape_inference.h379 const Shape& smaller_shape, const Shape& larger_shape,