Home
last modified time | relevance | path

Searched defs:inserted_window_dims (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc67 std::vector<int64_t> inserted_window_dims( in ConvertScatterDimensionNumbers() local
Dmlir_hlo_to_hlo.cc326 auto inserted_window_dims = ConvertDenseIntAttr(input.inserted_window_dims()); in Convert_scatter_dimension_numbers() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc1010 std::vector<int64> inserted_window_dims(dnums.inserted_window_dims().begin(), in ScatterOutputShardingFromUpdate() local
1034 std::vector<int64> inserted_window_dims(dnums.inserted_window_dims().begin(), in ScatterUpdateShardingFromOutput() local
Dhlo_instructions.cc2838 string inserted_window_dims = StrCat( in ScatterDimensionNumbersToString() local
2857 absl::Span<const int64> inserted_window_dims, in MakeScatterDimNumbers()
Dhlo_parser.cc2489 optional<std::vector<int64>> inserted_window_dims; in ParseInstructionRhs() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto508 repeated int64 inserted_window_dims = 2; field
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1118 auto inserted_window_dims = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4901 SmallVector<int64_t, 1> inserted_window_dims(1, 0); in matchAndRewrite() local