Home
last modified time | relevance | path

Searched refs:unique_indices (Results 1 – 25 of 35) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dunique_op.cc178 xla::XlaOp unique_indices = xla::Ne(shifted, sorted_data); in DataOutputFastPath() local
182 unique_indices = xla::DynamicUpdateSlice( in DataOutputFastPath()
183 unique_indices, true_r1, {xla::Zero(ctx->builder(), xla::S32)}); in DataOutputFastPath()
184 unique_indices = xla::ConvertElementType(unique_indices, xla::S32); in DataOutputFastPath()
188 std::vector<xla::XlaOp> to_sort_reverse = {sorted_iota, unique_indices}; in DataOutputFastPath()
197 unique_indices = xla::GetTupleElement(sort_reverse, 1); in DataOutputFastPath()
199 std::vector<xla::XlaOp> to_sort_final = {unique_indices, input}; in DataOutputFastPath()
209 unique_indices, xla::Zero(ctx->builder(), xla::S32), in DataOutputFastPath()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dgpu_ops.mlir18 // CHECK: unique_indices = false
30 … unique_indices = false} : (tensor<3x3xi32>, tensor<2xi32>, tensor<2x3xi32>) -> tensor<3x3xi32>
Dhlo_text_to_lhlo_no_opt.hlotxt36 // CHECK: unique_indices = false
/external/tensorflow/tensorflow/python/training/
Doptimizer.py77 unique_indices, new_index_positions = array_ops.unique(indices)
80 array_ops.shape(unique_indices)[0])
81 return (summed_values, unique_indices)
978 summed_grad, unique_indices = _deduplicate_indexed_slices(
980 return self._resource_apply_sparse(summed_grad, handle, unique_indices)
1030 summed_values, unique_indices = _deduplicate_indexed_slices(
1033 indices=unique_indices,
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc104 std::unordered_set<int> unique_indices; in ResolveAxis() local
110 if (unique_indices.find(current) == unique_indices.end()) { in ResolveAxis()
111 unique_indices.insert(current); in ResolveAxis()
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py82 unique_indices, new_index_positions = array_ops.unique(indices)
85 array_ops.shape(unique_indices)[0])
86 return (summed_values, unique_indices)
1273 summed_grad, unique_indices = _deduplicate_indexed_slices(
1275 return self._resource_apply_sparse(summed_grad, handle, unique_indices,
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dcanonicalize.mlir1259 unique_indices = false
1282 unique_indices = false
1306 unique_indices = false
1330 unique_indices = false
1354 unique_indices = false
1377 unique_indices = false
1402 unique_indices = false
1427 unique_indices = false
1451 unique_indices = false
1472 unique_indices = false
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_cluster_formation.cc309 llvm::SmallDenseSet<int64_t, 8> unique_indices; in SortTPUReplicatedInputsByIndex() local
316 if (!unique_indices.insert(index).second) in SortTPUReplicatedInputsByIndex()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h188 bool unique_indices) override;
Dmlir_hlo_builder.cc304 bool unique_indices) { in ScatterInternal() argument
311 builder_.getBoolAttr(unique_indices)); in ScatterInternal()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc2821 bool unique_indices) in HloScatterInstruction() argument
2824 unique_indices_(unique_indices) { in HloScatterInstruction()
2879 proto.set_unique_indices(unique_indices()); in ToProto()
2890 if (unique_indices()) { in ExtraAttributesToStringImpl()
2906 unique_indices() == casted_other.unique_indices(); in IdenticalSlowPath()
2915 scatter_dimension_numbers(), indices_are_sorted(), unique_indices()); in CloneWithNewOperandsImpl()
Dhlo_instruction.h954 bool indices_are_sorted, bool unique_indices);
1830 virtual bool unique_indices() const { LOG(FATAL) << "Unimplemented method."; } in unique_indices() function
Dhlo_instructions.h1705 bool indices_are_sorted, bool unique_indices);
1714 bool unique_indices() const override { return unique_indices_; } in unique_indices() function
Dhlo.proto252 bool unique_indices = 69; field
Dhlo_instruction.cc691 proto.indices_are_sorted(), proto.unique_indices()); in CreateFromProto()
1629 bool unique_indices) { in CreateScatter() argument
1632 scatter_dim_numbers, indices_are_sorted, unique_indices); in CreateScatter()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc626 scatter->unique_indices())); in HandleScatter()
658 scatter->indices_are_sorted(), scatter->unique_indices())); in HandleScatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h483 bool unique_indices; member
Dir_emitter_unnested.cc1966 desc.unique_indices = root->unique_indices(); in HandleFusion()
2492 if (!scatter->unique_indices()) { in HandleScatter()
2585 desc.unique_indices = scatter.unique_indices(); in EmitScatter()
2596 if (!desc.unique_indices) { in EmitScatter()
2709 if (!desc.unique_indices) { in EmitScatter()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h859 bool indices_are_sorted = false, bool unique_indices = false);
865 bool unique_indices);
1399 bool indices_are_sorted, bool unique_indices);
2450 bool indices_are_sorted = false, bool unique_indices = false);
Dxla_builder.cc2305 bool indices_are_sorted, bool unique_indices) { in Scatter() argument
2319 indices_are_sorted, unique_indices); in Scatter()
2327 bool unique_indices) { in ScatterInternal() argument
2331 instr.set_unique_indices(unique_indices); in ScatterInternal()
4885 bool indices_are_sorted, bool unique_indices) { in Scatter() argument
4888 indices_are_sorted, unique_indices); in Scatter()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir209 // CHECK-SAME: unique_indices = false
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlegalize_hlo.mlir1954 unique_indices = false} : (tensor<20x6xf32>, tensor<4xi32>, tensor<4x6xf32>) -> tensor<20x6xf32>
1977 … unique_indices = false} : (tensor<20x6xf32>, tensor<4x1xi32>, tensor<4x6xf32>) -> tensor<20x6xf32>
2000 … unique_indices = false} : (tensor<20x6xf32>, tensor<4x1xi32>, tensor<4x6xf32>) -> tensor<20x6xf32>
2023 … unique_indices = false} : (tensor<20x6xf32>, tensor<4x1xi32>, tensor<4x6xf32>) -> tensor<20x6xf32>
2046 … unique_indices = false} : (tensor<20x6xf32>, tensor<4x1xi32>, tensor<4x6xf32>) -> tensor<20x6xf32>
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td474 DefaultValuedAttr<BoolAttr, "false">:$unique_indices
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir770 unique_indices = true
781 …_to_operand_dims={0,1}, index_vector_dim=1, indices_are_sorted=true, unique_indices=true, to_apply…
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc440 "unique_indices", builder_->getBoolAttr(scatter->unique_indices()))); in ImportInstructionImpl()

12