Searched refs:ScatterDimensionNumbers (Results 1 – 22 of 22) sorted by relevance
133 HloInstruction* index_vector, const ScatterDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace()222 const ScatterDimensionNumbers& dim_numbers = in ScatterLoopBody()350 const ScatterDimensionNumbers& dim_numbers = in ExpandScatter()
1405 const ScatterDimensionNumbers& scatter_dim_numbers);1406 const ScatterDimensionNumbers& scatter_dimension_numbers() const { in scatter_dimension_numbers()1416 static ScatterDimensionNumbers MakeScatterDimNumbers(1434 std::unique_ptr<ScatterDimensionNumbers> scatter_dimension_numbers_;
293 const ScatterDimensionNumbers& scatter_dim_numbers);
2059 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in IterationSpaceForUpdateScatterIndices()2078 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in IterationSpaceForUpdateWindowIndices()2106 const ScatterDimensionNumbers* dim_numbers, const Shape& input_shape, in UpdateScatterIndexToInputIndex()2220 const ScatterDimensionNumbers& dim_numbers_;2236 const ScatterDimensionNumbers& dim_numbers, const Shape& input_shape, in UpdateWindowIndexToInputIndex()2308 const ScatterDimensionNumbers& dim_numbers = in HandleScatter()
735 const ScatterDimensionNumbers& scatter_dim_numbers);1598 const ScatterDimensionNumbers& scatter_dimension_numbers() const;
2398 const ScatterDimensionNumbers& scatter_dim_numbers) in HloScatterInstruction()2405 absl::make_unique<ScatterDimensionNumbers>(scatter_dim_numbers); in HloScatterInstruction()2428 /* static */ ScatterDimensionNumbers2434 ScatterDimensionNumbers scatter_dim_numbers; in MakeScatterDimNumbers()
677 ScatterDimensionNumbers dim_numbers; in TEST_F()
185 xla.ScatterDimensionNumbers scatter_dimension_numbers = 48;
543 absl::make_unique<ScatterDimensionNumbers>( in CreateFromProto()1318 const ScatterDimensionNumbers& scatter_dim_numbers) { in CreateScatter()3524 const ScatterDimensionNumbers& HloInstruction::scatter_dimension_numbers() in scatter_dimension_numbers()
3071 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) { in ValidateScatterDimensionNumbers()3166 const ScatterDimensionNumbers& scatter_dim_numbers) { in InferScatterShape()
1672 ScatterDimensionNumbers dim_numbers = in ParseInstructionRhs()
138 xla::ScatterDimensionNumbers dim_numbers; in XlaScatter()
605 %typemap(in) const ScatterDimensionNumbers&606 (ScatterDimensionNumbers dimension_numbers) {
379 const ScatterDimensionNumbers& dimension_numbers);
765 const ScatterDimensionNumbers& dimension_numbers) { in Scatter()
1961 class ScatterDimensionNumbers(object): class
1683 dnums = xla_client.ScatterDimensionNumbers()
170 xla::ScatterDimensionNumbers scatter_dim_numbers; in ScatterToGradData()
549 const ScatterDimensionNumbers& dimension_numbers);974 const ScatterDimensionNumbers& dimension_numbers);1822 const ScatterDimensionNumbers& dimension_numbers);
1867 const ScatterDimensionNumbers& dimension_numbers) { in Scatter()3433 const ScatterDimensionNumbers& dimension_numbers) { in Scatter()
469 message ScatterDimensionNumbers { message
1081 const ScatterDimensionNumbers& dim_numbers = in EmitScatter()