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.cc350 auto inserted_window_dims = ConvertDenseIntAttr(input.inserted_window_dims()); in Convert_scatter_dimension_numbers() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc1053 std::vector<int64> inserted_window_dims(dnums.inserted_window_dims().begin(), in ScatterOutputShardingFromUpdate() local
1077 std::vector<int64> inserted_window_dims(dnums.inserted_window_dims().begin(), in ScatterUpdateShardingFromOutput() local
Dhlo_instructions.cc3002 string inserted_window_dims = StrCat( in ScatterDimensionNumbersToString() local
3021 absl::Span<const int64> inserted_window_dims, in MakeScatterDimNumbers()
Dhlo_parser.cc2645 optional<std::vector<int64>> inserted_window_dims; in ParseInstructionRhs() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla_data.proto514 repeated int64 inserted_window_dims = 2; field
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1793 auto inserted_window_dims = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6016 SmallVector<int64_t, 1> inserted_window_dims(1, 0); in matchAndRewrite() local