Home
last modified time | relevance | path

Searched defs:start_indices_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dgather_expander.cc33 const Shape& start_indices_shape = start_indices->shape(); in TransposeIndexVectorDimToLast() local
89 const Shape& start_indices_shape, HloInstruction* accumulator, in AdjustBatchDimsInAccumulator()
277 const Shape& start_indices_shape = start_indices->shape(); in GatherLoopTripCount() local
H A Dshape_inference.cc2604 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicSliceShape() local
2709 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape() local
3331 const Shape& input_shape, absl::Span<const int64_t> start_indices_shape, in ValidateGatherDimensionNumbers()
3422 const Shape& input_shape, const Shape& start_indices_shape, in InferGatherShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Ddynamic_ops_test.cc778 auto start_indices_shape = ShapeUtil::MakeShape(S32, {}); in BM_DynamicSlice() local