Home
last modified time | relevance | path

Searched refs:larger_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dimage_grad_test_base.py117 def _itGen(self, smaller_shape, larger_shape): argument
118 up_sample = (smaller_shape, larger_shape)
119 down_sample = (larger_shape, smaller_shape)
120 pass_through = (larger_shape, larger_shape)
160 larger_shape = [batch_size, 4, 6, channel_count]
161 for in_shape, out_shape, _, _ in self._itGen(smaller_shape, larger_shape):
191 larger_shape = [batch_size, 5, 6, channel_count]
193 self._itGen(smaller_shape, larger_shape):
262 larger_shape = [batch_size, 8, 16, channel_count]
263 for params in self._itGen(smaller_shape, larger_shape):
/external/ComputeLibrary/tests/validation/NEON/
DArithmeticSubtraction.cpp123 const auto larger_shape = TensorShape{ 27U, 13U, 2U }; in TEST_CASE()
126 const auto larger_tensor_info = TensorInfo{ larger_shape, 1, DataType::F32 }; in TEST_CASE()
135 const auto larger_shape = TensorShape{ 27U, 13U, 2U }; in TEST_CASE()
138 const auto larger_tensor_info = TensorInfo{ larger_shape, 1, DataType::F32 }; in TEST_CASE()
DPixelWiseMultiplication.cpp234 const auto larger_shape = TensorShape{ 27U, 13U, 2U }; in TEST_CASE()
237 const auto larger_tensor_info = TensorInfo{ larger_shape, 1, DataType::F32 }; in TEST_CASE()
246 const auto larger_shape = TensorShape{ 27U, 13U, 2U }; in TEST_CASE()
249 const auto larger_tensor_info = TensorInfo{ larger_shape, 1, DataType::F32 }; in TEST_CASE()
/external/ComputeLibrary/tests/validation/CL/
DArithmeticSubtraction.cpp99 const auto larger_shape = TensorShape{ 27U, 13U, 2U }; in TEST_CASE()
102 const auto larger_tensor_info = TensorInfo{ larger_shape, 1, DataType::F32 }; in TEST_CASE()
111 const auto larger_shape = TensorShape{ 27U, 13U, 2U }; in TEST_CASE()
114 const auto larger_tensor_info = TensorInfo{ larger_shape, 1, DataType::F32 }; in TEST_CASE()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc783 const Shape& smaller_shape, const Shape& larger_shape, in InferInDimBroadcastShape() argument
791 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()
839 Shape output_shape(larger_shape); in InferInDimBroadcastShape()
841 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()
850 if (dimension_to_match >= larger_shape.dimensions_size()) { in InferInDimBroadcastShape()
854 dimension_to_match, larger_shape.dimensions_size()); in InferInDimBroadcastShape()
857 int64_t large_dimension_size = larger_shape.dimensions(dimension_to_match); in InferInDimBroadcastShape()
860 larger_shape.is_dynamic_dimension(dimension_to_match); in InferInDimBroadcastShape()
870 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()
881 ShapeUtil::HumanString(larger_shape)); in InferInDimBroadcastShape()
[all …]
Dshape_inference.h407 const Shape& smaller_shape, const Shape& larger_shape,