| /external/deqp-deps/SPIRV-Tools/source/fuzz/ |
| D | transformation_permute_function_parameters.cpp | 68 std::unordered_set<uint32_t> unique_indices; in IsApplicable() local
75 unique_indices.insert(index); in IsApplicable()
79 assert(unique_indices.size() == arg_size && "Permutation has duplicates"); in IsApplicable()
|
| /external/angle/third_party/spirv-tools/src/source/fuzz/ |
| D | transformation_permute_function_parameters.cpp | 68 std::unordered_set<uint32_t> unique_indices; in IsApplicable() local
75 unique_indices.insert(index); in IsApplicable()
79 assert(unique_indices.size() == arg_size && "Permutation has duplicates"); in IsApplicable()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 76 unique_indices, new_index_positions = array_ops.unique(indices)
79 array_ops.shape(unique_indices)[0])
80 return (summed_values, unique_indices)
980 summed_grad, unique_indices = _deduplicate_indexed_slices(
982 return self._resource_apply_sparse(summed_grad, handle, unique_indices)
1032 summed_values, unique_indices = _deduplicate_indexed_slices(
1035 indices=unique_indices,
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | optimizer_v2.py | 70 unique_indices, new_index_positions = array_ops.unique(indices)
73 array_ops.shape(unique_indices)[0])
74 return (summed_values, unique_indices)
953 summed_grad, unique_indices = _deduplicate_indexed_slices(
955 return self._resource_apply_sparse(summed_grad, handle, unique_indices,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 2537 bool unique_indices) in HloScatterInstruction() argument
2540 unique_indices_(unique_indices) { in HloScatterInstruction()
2595 proto.set_unique_indices(unique_indices()); in ToProto()
2606 if (unique_indices()) { in ExtraAttributesToStringImpl()
2622 unique_indices() == casted_other.unique_indices(); in IdenticalSlowPath()
2631 scatter_dimension_numbers(), indices_are_sorted(), unique_indices()); in CloneWithNewOperandsImpl()
|
| D | hlo_instruction.h | 883 bool indices_are_sorted, bool unique_indices);
1722 virtual bool unique_indices() const { LOG(FATAL) << "Unimplemented method."; } in unique_indices() function
|
| D | hlo_instructions.h | 1499 bool indices_are_sorted, bool unique_indices);
1508 bool unique_indices() const override { return unique_indices_; } in unique_indices() function
|
| D | hlo.proto | 243 bool unique_indices = 69; field
|
| D | hlo_parser.cc | 1758 optional<bool> unique_indices = false; in ParseInstructionRhs() local
1760 &unique_indices}; in ParseInstructionRhs()
1777 indices_are_sorted.value(), unique_indices.value())); in ParseInstructionRhs()
|
| D | hlo_instruction.cc | 586 proto.indices_are_sorted(), proto.unique_indices()); in CreateFromProto()
1431 bool unique_indices) { in CreateScatter() argument
1434 scatter_dim_numbers, indices_are_sorted, unique_indices); in CreateScatter()
|
| D | hlo_parser_test.cc | 919 …ms={}, scatter_dims_to_operand_dims={0,1,2,3,4}, index_vector_dim=4, unique_indices=true, to_apply…
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1626 unique_indices=False): argument
1630 indices_are_sorted, unique_indices)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 590 bool indices_are_sorted = false, bool unique_indices = false);
1011 bool indices_are_sorted, bool unique_indices);
1891 bool indices_are_sorted = false, bool unique_indices = false);
|
| D | xla_builder.cc | 1805 bool indices_are_sorted, bool unique_indices) { in Scatter() argument
1810 instr.set_unique_indices(unique_indices); in Scatter()
3500 bool indices_are_sorted, bool unique_indices) { in Scatter() argument
3503 indices_are_sorted, unique_indices); in Scatter()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 681 unique_indices = true
692 …_to_operand_dims={0,1}, index_vector_dim=1, indices_are_sorted=true, unique_indices=true, to_apply…
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 706 op.indices_are_sorted(), op.unique_indices()); in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | hlo_ops.td | 1022 DefaultValuedAttr<BoolAttr, "false">:$unique_indices
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1392 | `unique_indices` | `bool` | Whether the indices are |
1463 If `unique_indices` is set to true then XLA can assume that all element
1465 `unique_indices` is set to true and the indices being scattered to are not
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 2940 …<0> : tensor<1xi64>, update_window_dims = dense<2> : tensor<1xi64>}, unique_indices = false} : (te…
2957 …<0> : tensor<1xi64>, update_window_dims = dense<2> : tensor<1xi64>}, unique_indices = false} : (te…
3067 // CHECK-SAME: unique_indices = false
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1001 if (!scatter->unique_indices()) { in EmitScatter()
|