Home
last modified time | relevance | path

Searched defs:output_shape_size (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Dcollective_lowering.cc411 int64 output_shape_size) { in SelectElementsBasedOnId()
490 const int64 output_shape_size = output_shape.size(); in LowerAllGatherOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc440 int64_t rhs_shape_size, int64_t lhs_shape_size, int64_t output_shape_size, in GetWindowedEinsumConfiguration()