Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc840 int64 dimension_to_match = broadcast_dimensions.at(i); in InferInDimBroadcastShape() local
841 if (dimension_to_match < 0) { in InferInDimBroadcastShape()
844 dimension_to_match); 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()
882 if (i > 0 && broadcast_dimensions.at(i - 1) >= dimension_to_match) { in InferInDimBroadcastShape()
885 dimension_to_match, broadcast_dimensions.at(i - 1)); in InferInDimBroadcastShape()
888 output_shape.set_dimensions(dimension_to_match, small_dimension_size); in InferInDimBroadcastShape()
[all …]