Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc1136 std::vector<int64> reshaped_dimensions; in CreateBroadcastSequence() local
1140 reshaped_dimensions.push_back(operand->shape().dimensions(i)); in CreateBroadcastSequence()
1151 reshaped_dimensions), in CreateBroadcastSequence()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc449 std::vector<int64> reshaped_dimensions; in AddBroadcastSequence() local
453 reshaped_dimensions.push_back(operand_shape.dimensions(i)); in AddBroadcastSequence()
464 reshaped_dimensions), in AddBroadcastSequence()