Home
last modified time | relevance | path

Searched defs:scatter_indices_shape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dscatter_expander.cc36 const Shape& scatter_indices_shape = scatter_indices->shape(); in TransposeIndexVectorDimToLast() local
118 const Shape& scatter_indices_shape, HloInstruction* updates, in AdjustScatterDims()
383 const Shape& scatter_indices_shape = scatter_indices->shape(); in ScatterTripCount() local
H A Dshape_inference.cc3564 const Shape& operand_shape, absl::Span<const int64_t> scatter_indices_shape, in ValidateScatterDimensionNumbers()
3668 const Shape& scatter_indices_shape = *arg_shapes[operand_count]; in InferScatterShape() local
H A Dshape_inference_test.cc2972 const Shape& scatter_indices_shape, in CreateShapes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.h483 Shape scatter_indices_shape; member