Searched defs:start_indices_shape (Results 1 – 3 of 3) sorted by relevance
33 const Shape& start_indices_shape = start_indices->shape(); in TransposeIndexVectorDimToLast() local89 const Shape& start_indices_shape, HloInstruction* accumulator, in AdjustBatchDimsInAccumulator()277 const Shape& start_indices_shape = start_indices->shape(); in GatherLoopTripCount() local
2604 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicSliceShape() local2709 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape() local3331 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()
778 auto start_indices_shape = ShapeUtil::MakeShape(S32, {}); in BM_DynamicSlice() local