Home
last modified time | relevance | path

Searched defs:unique_indices (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_scatter_gather_ops.py26 def _fill_indices(self, idx, dim, dim_size, elems_per_row, m, n, o, unique_indices=True): argument
87 unique_indices=True, include_self=True): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.h486 bool unique_indices; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DEmbedding.cu360 auto unique_indices = at::empty(indices.numel(), indices.options()); in embedding_renorm_cuda_() local
/aosp_15_r20/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
H A Dlayer_norm.cc104 std::unordered_set<int> unique_indices; in ResolveAxis() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo.proto311 bool unique_indices = 69; field
H A Dhlo_instruction.h2046 virtual bool unique_indices() const { LOG(FATAL) << "Unimplemented method."; } in unique_indices() function
H A Dhlo_instructions.h2164 bool unique_indices() const override { return unique_indices_; } in unique_indices() function
H A Dhlo_instruction.cc1940 bool unique_indices) { in CreateScatter()
1951 bool unique_indices) { in CreateScatter()
H A Dhlo_parser.cc2867 optional<bool> unique_indices = false; in CreateInstruction() local
H A Dhlo_instructions.cc3144 bool unique_indices) in HloScatterInstruction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc2430 bool indices_are_sorted, bool unique_indices) { in Scatter()
2440 bool indices_are_sorted, bool unique_indices) { in Scatter()
2478 bool unique_indices) { in ScatterInternal()
4909 bool indices_are_sorted, bool unique_indices) { in Scatter()
4919 bool indices_are_sorted, bool unique_indices) { in Scatter()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
H A Dmlir_hlo_builder.cc398 bool unique_indices) { in ScatterInternal()