Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dscatter_expander.cc33 const Shape& scatter_indices_shape = scatter_indices->shape(); in TransposeIndexVectorDimToLast() local
115 const Shape& scatter_indices_shape, HloInstruction* updates, in AdjustScatterDims()
332 const Shape& scatter_indices_shape = scatter_indices->shape(); in ScatterTripCount() local
Dshape_inference.cc3582 const Shape& operand_shape, absl::Span<const int64> scatter_indices_shape, in ValidateScatterDimensionNumbers()
3676 const Shape& operand_shape, const Shape& scatter_indices_shape, in InferScatterShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h410 Shape scatter_indices_shape; member