Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc112 int64 operand_rank) { in ExpandIndexVectorIntoOperandSpace() argument
116 if (operand_rank == 0) { in ExpandIndexVectorIntoOperandSpace()
131 for (int i = 0; i < operand_rank; i++) { in ExpandIndexVectorIntoOperandSpace()
Dscatter_expander.cc134 int64 operand_rank) { in ExpandIndexVectorIntoOperandSpace() argument
139 if (operand_rank == 0) { in ExpandIndexVectorIntoOperandSpace()
152 for (int i = 0; i < operand_rank; i++) { in ExpandIndexVectorIntoOperandSpace()
Dshape_inference.cc2630 const int64 operand_rank = operand_shape.rank(); in InferBroadcastShape() local
2632 if (operand_rank > output_rank) { in InferBroadcastShape()
2636 operand_rank, output_rank); in InferBroadcastShape()
2638 if (operand_rank != broadcast_dimensions.size()) { in InferBroadcastShape()
2642 operand_rank, broadcast_dimensions.size()); in InferBroadcastShape()
2644 for (int64 i = 0; i < operand_rank; i++) { in InferBroadcastShape()
Dlayout_assignment.cc1270 int64 operand_rank = operand->shape().rank(); in PropagateOperandConstraint() local
1271 if (operand_rank <= 1) { in PropagateOperandConstraint()
1289 if (operand_rank != sibling_rank) { in PropagateOperandConstraint()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc662 const int64 operand_rank = operand_shape.rank(); in Broadcast() local
663 std::vector<int64> dimensions(operand_rank); in Broadcast()
664 for (int i = 0; i < operand_rank; ++i) { in Broadcast()
665 dimensions[i] = i + shape.rank() - operand_rank; in Broadcast()