Searched refs:smaller_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 744 const Shape& smaller_shape, const Shape& larger_shape, in InferInDimBroadcastShape() argument
746 if (broadcast_dimensions.empty() && !ShapeUtil::IsScalar(smaller_shape)) { in InferInDimBroadcastShape()
751 ShapeUtil::HumanString(smaller_shape), in InferInDimBroadcastShape()
753 } else if (broadcast_dimensions.size() != smaller_shape.rank()) { in InferInDimBroadcastShape()
759 smaller_shape.rank(), broadcast_dimensions.size()); in InferInDimBroadcastShape()
802 ShapeUtil::HigherPrecisionElementType(larger_shape, smaller_shape)); in InferInDimBroadcastShape()
804 for (int i = 0; i < smaller_shape.dimensions_size(); ++i) { in InferInDimBroadcastShape()
817 int64 small_dimension_size = smaller_shape.dimensions(i); in InferInDimBroadcastShape()
819 bool small_is_dynamic = smaller_shape.is_dynamic_dimension(i); in InferInDimBroadcastShape()
830 ShapeUtil::HumanString(smaller_shape), in InferInDimBroadcastShape()
[all …]
|
| D | shape_inference.h | 333 const Shape& smaller_shape, const Shape& larger_shape,
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_test_util.py | 810 smaller_shape = batch_shape + [m, n - 1]
821 smaller_shape, mean=0.0, stddev=stddev_mat, dtype=dtype, seed=seed)
|