Home
last modified time | relevance | path

Searched refs:ScatterDimensionNumbers (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.h37 mlir::mhlo::ScatterDimensionNumbers ConvertScatterDimensionNumbers(
38 const xla::ScatterDimensionNumbers& dnums, mlir::Builder* builder);
Dattribute_importer.cc63 mlir::mhlo::ScatterDimensionNumbers ConvertScatterDimensionNumbers( in ConvertScatterDimensionNumbers()
64 const xla::ScatterDimensionNumbers& dnums, mlir::Builder* builder) { in ConvertScatterDimensionNumbers()
72 return mlir::mhlo::ScatterDimensionNumbers::get( in ConvertScatterDimensionNumbers()
Dmlir_hlo_to_hlo.cc317 static xla::ScatterDimensionNumbers Convert_scatter_dimension_numbers( in Convert_scatter_dimension_numbers()
318 mlir::mhlo::ScatterDimensionNumbers input) { in Convert_scatter_dimension_numbers()
319 xla::ScatterDimensionNumbers output; in Convert_scatter_dimension_numbers()
947 xla::ScatterDimensionNumbers dimension_numbers = in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/xla/service/
Dscatter_expander.cc133 HloInstruction* index_vector, const ScatterDimensionNumbers& dim_numbers, in ExpandIndexVectorIntoOperandSpace()
222 const ScatterDimensionNumbers& dim_numbers = in ScatterLoopBody()
333 const ScatterDimensionNumbers& dim_numbers = in ScatterTripCount()
366 const ScatterDimensionNumbers& dim_numbers = in ExpandInstruction()
Dhlo_instructions.h1704 const ScatterDimensionNumbers& scatter_dim_numbers,
1706 const ScatterDimensionNumbers& scatter_dimension_numbers() const { in scatter_dimension_numbers()
1719 static ScatterDimensionNumbers MakeScatterDimNumbers(
1726 const ScatterDimensionNumbers& scatter_dimension_numbers);
1740 std::unique_ptr<ScatterDimensionNumbers> scatter_dimension_numbers_;
Dshape_inference.h324 const ScatterDimensionNumbers& scatter_dim_numbers);
Dhlo_sharding_util.cc645 const ScatterDimensionNumbers& dnums = hlo->scatter_dimension_numbers(); in ScatterIndexSharding()
678 const ScatterDimensionNumbers& dnums = hlo->scatter_dimension_numbers(); in ScatterDataSharding()
713 const ScatterDimensionNumbers& dnums = hlo.scatter_dimension_numbers(); in ScatterEffectiveIndexSharding()
756 const ScatterDimensionNumbers& dnums = hlo.scatter_dimension_numbers(); in ScatterEffectiveDataSharding()
Dhlo_evaluator_typed_visitor.h2082 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) {
2101 const Shape& updates_shape, const ScatterDimensionNumbers& dim_numbers) {
2129 const ScatterDimensionNumbers* dim_numbers, const Shape& input_shape,
2243 const ScatterDimensionNumbers& dim_numbers_;
2259 const ScatterDimensionNumbers& dim_numbers, const Shape& input_shape,
2331 const ScatterDimensionNumbers& dim_numbers =
Dhlo_instructions.cc2820 const ScatterDimensionNumbers& scatter_dim_numbers, bool indices_are_sorted, in HloScatterInstruction()
2830 absl::make_unique<ScatterDimensionNumbers>(scatter_dim_numbers); in HloScatterInstruction()
2834 const ScatterDimensionNumbers& scatter_dimension_numbers) { in ScatterDimensionNumbersToString()
2854 /* static */ ScatterDimensionNumbers
2860 ScatterDimensionNumbers scatter_dim_numbers; in MakeScatterDimNumbers()
Dhlo_instruction.h953 const ScatterDimensionNumbers& scatter_dim_numbers,
1888 const ScatterDimensionNumbers& scatter_dimension_numbers() const;
Dhlo.proto194 xla.ScatterDimensionNumbers scatter_dimension_numbers = 48;
Dhlo_cost_analysis_test.cc1009 ScatterDimensionNumbers dim_numbers; in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops_base_structs.td32 def ScatterDimensionNumbers : StructAttr<
33 "ScatterDimensionNumbers", HLO_Dialect, [
Dlhlo_ops.td472 ScatterDimensionNumbers:$scatter_dimension_numbers,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_scatter_ops.cc96 xla::ScatterDimensionNumbers dnums_;
Dresampler_ops.cc170 xla::ScatterDimensionNumbers scatter_dim_numbers; in ScatterToGradData()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc138 xla::ScatterDimensionNumbers dim_numbers; in XlaScatter()
/external/tensorflow/tensorflow/compiler/xla/python/
Dtypes.h318 struct type_caster<xla::ScatterDimensionNumbers> {
320 PYBIND11_TYPE_CASTER(xla::ScatterDimensionNumbers,
Dxla_client.py651 class ScatterDimensionNumbers(object): class
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.h54 xla::StatusOr<mhlo::ScatterDimensionNumbers> GetScatterDimensionNumbers(
Dmhlo_to_lhlo_with_xla.cc543 StatusOr<mhlo::ScatterDimensionNumbers>
547 const xla::ScatterDimensionNumbers& xla_scatter_dim = in GetScatterDimensionNumbers()
549 auto scatter_dimension_numbers = mhlo::ScatterDimensionNumbers::get( in GetScatterDimensionNumbers()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h187 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted,
Dmlir_hlo_builder.cc303 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted, in ScatterInternal()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h482 mlir::mhlo::ScatterDimensionNumbers dim_numbers;
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h858 const ScatterDimensionNumbers& dimension_numbers,
864 const ScatterDimensionNumbers& dimension_numbers, bool indices_are_sorted,
1398 const ScatterDimensionNumbers& dimension_numbers,
2449 const ScatterDimensionNumbers& dimension_numbers,

12