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()275 const Shape& start_indices_shape = start_indices->shape(); in GatherLoopTripCount() local
2487 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicSliceShape() local2592 const Shape& start_indices_shape = start_index_shapes[0]; in InferDynamicUpdateSliceShape() local3212 const Shape& input_shape, absl::Span<const int64> start_indices_shape, in ValidateGatherDimensionNumbers()3303 const Shape& input_shape, const Shape& start_indices_shape, in InferGatherShape()
87 const TensorShape start_indices_shape = ctx->InputShape("start_indices"); in Compile() local
775 auto start_indices_shape = ShapeUtil::MakeShape(S32, {}); in BM_DynamicSlice() local