| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_tensorflow_merge.cc | 32 const auto* merge_op = merge_it->get(); in Run() local
33 if (merge_op->type != OperatorType::kMerge) { in Run()
41 if (merge_op->inputs.size() > 1) { in Run()
42 AddMessageF("Waiting for %s to be resolved", LogName(*merge_op)); in Run()
48 CHECK_EQ(merge_op->inputs.size(), 1); in Run()
53 if (input == merge_op->outputs[0]) { in Run()
54 input = merge_op->inputs[0]; in Run()
59 DeleteOpAndArrays(model, merge_op); in Run()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 41 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp() argument
42 if (merge_op == "Add") { in ReductionOp()
45 } else if (merge_op == "Mul") { in ReductionOp()
48 } else if (merge_op == "Maximum") { in ReductionOp()
51 } else if (merge_op == "Minimum") { in ReductionOp()
57 merge_op); in ReductionOp()
139 ReductionOp(col_params->merge_op->type_string(), &reduction_op); in Enqueue()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 27 merge_op='Add', argument
67 merge_op=merge_op,
78 merge_op='Add', argument
116 merge_op=merge_op,
|
| D | collective_ops_gpu_test.py | 254 def run_all_reduce(group_key, instance_key, merge_op): argument
260 in0, self._group_size, group_key, instance_key, merge_op,
265 in1, self._group_size, group_key, instance_key, merge_op,
271 merge_op = combination[0]
272 results = run_all_reduce(group_key=10, instance_key=20, merge_op=merge_op)
284 [1.], group_size=1, group_key=0, instance_key=0, merge_op='Add',
|
| D | collective_ops_test.py | 56 merge_op='Add', argument
80 merge_op,
289 merge_op='Max',
300 merge_op='Min',
468 merge_op='Add', final_op='Id')
473 merge_op='Add', final_op='Id')
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectiveReduce.pbtxt | 37 name: "merge_op"
100 name: "merge_op"
171 name: "merge_op"
249 name: "merge_op"
|
| D | CollectiveReduceV2.pbtxt | 37 name: "merge_op"
103 name: "merge_op"
181 name: "merge_op"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | control_flow_upgrade_legacy_v1.py | 41 merge_op = control_flow_ops.merge(switch_op)[0]
42 result = tf.transpose(merge_op)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CollectiveReduce.pbtxt | 37 name: "merge_op"
100 name: "merge_op"
171 name: "merge_op"
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 397 const std::string merge_op("Add"); in CollectiveSum() local
398 TFE_OpSetAttrString(op.get(), "merge_op", merge_op.c_str(), in CollectiveSum()
399 merge_op.length()); in CollectiveSum()
488 const std::string merge_op("Mul"); in RegisterCollectiveMulFunction() local
489 TF_SetAttrString(reduce_desc, "merge_op", merge_op.c_str(), in RegisterCollectiveMulFunction()
490 merge_op.length()); in RegisterCollectiveMulFunction()
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_lower_alu_to_scalar.c | 65 lower_reduction(nir_alu_instr *alu, nir_op chan_op, nir_op merge_op, in lower_reduction() argument
88 last = nir_build_alu(builder, merge_op, in lower_reduction()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 408 << ((col_params_->merge_op != nullptr) ? "tmp_chunk" : "chunk"); in DispatchRecv()
409 Tensor* dst_tensor = (!rf->second_pass && (col_params_->merge_op != nullptr)) in DispatchRecv()
|
| D | ring_reducer.cc | 259 col_params_->merge_op, &rf->chunk, &rf->tmp_chunk); in RunAsyncParts()
|
| D | ring_reducer_test.cc | 479 col_params_->merge_op = merge_op_.get(); in DoReduce()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond_test.cc | 134 auto merge_op = ops::Merge( in TEST() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_multi_worker_test.py | 118 merge_op="Add",
|
| D | control_flow_ops_py_test.py | 240 merge_op = control_flow_ops.merge(switch_op)[0]
242 val = merge_op.values
243 ind = merge_op.indices
268 merge_op = control_flow_ops.merge(switch_op)[0]
270 result = self.evaluate(merge_op)
282 merge_op = control_flow_ops.merge([add_op, id_op])[0]
284 result = self.evaluate(merge_op)
297 merge_op = control_flow_ops.merge([add_op, mul_op])[0]
299 result = self.evaluate(merge_op)
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | folding_rules.cpp | 801 SpvOp merge_op = inst->opcode(); in MergeDivDivArithmetic() local
804 merge_op = SpvOpFMul; in MergeDivDivArithmetic()
811 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeDivDivArithmetic()
1246 SpvOp merge_op = inst->opcode(); in MergeSubSubArithmetic() local
1248 merge_op = uses_float ? SpvOpFAdd : SpvOpIAdd; in MergeSubSubArithmetic()
1253 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeSubSubArithmetic()
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | folding_rules.cpp | 801 SpvOp merge_op = inst->opcode(); in MergeDivDivArithmetic() local
804 merge_op = SpvOpFMul; in MergeDivDivArithmetic()
811 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeDivDivArithmetic()
1246 SpvOp merge_op = inst->opcode(); in MergeSubSubArithmetic() local
1248 merge_op = uses_float ? SpvOpFAdd : SpvOpIAdd; in MergeSubSubArithmetic()
1253 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeSubSubArithmetic()
|
| /external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/ |
| D | folding_rules.cpp | 854 SpvOp merge_op = inst->opcode(); in MergeDivDivArithmetic() local
857 merge_op = SpvOpFMul; in MergeDivDivArithmetic()
864 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeDivDivArithmetic()
1299 SpvOp merge_op = inst->opcode(); in MergeSubSubArithmetic() local
1301 merge_op = uses_float ? SpvOpFAdd : SpvOpIAdd; in MergeSubSubArithmetic()
1306 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeSubSubArithmetic()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 146 OpKernel* merge_op = nullptr; // reduction only member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 269 col_params_->merge_op = merge_op_.get(); in CollectiveReduceOpKernel()
535 col_params->merge_op = merge_op_.get(); in ComputeAsync()
|
| D | collective_nccl_test.cc | 474 di->col_params_->merge_op = di->merge_op_.get(); in InitDevice()
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device_test.py | 65 merge_op=operation,
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_Numpy_Distributed_Image_Classification.ipynb | 467 " instance_key=instance_key, merge_op='Add', final_op='Div',\n",
|