Searched defs:start_indices_shape (Results 1 – 4 of 4) sorted by relevance
31 const Shape& start_indices_shape = start_indices->shape(); in TransposeIndexVectorDimToLast() local87 const Shape& start_indices_shape, HloInstruction* accumulator, in AdjustBatchDimsInAccumulator()314 const Shape& start_indices_shape = start_indices->shape(); in ExpandInstruction() local
2279 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicSliceShape() local2384 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape() local2867 const Shape& input_shape, absl::Span<const int64> start_indices_shape, in ValidateGatherDimensionNumbers()2958 const Shape& input_shape, const Shape& start_indices_shape, in InferGatherShape()
76 const TensorShape start_indices_shape = ctx->InputShape("start_indices"); in Compile() local
758 auto start_indices_shape = ShapeUtil::MakeShape(S32, {}); in BM_DynamicSlice() local