| /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()
60 AddMessageF("Removing already-resolved %s", LogName(*merge_op)); in Run()
61 model->EraseArray(merge_op->outputs[0]); in Run()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer.cc | 61 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp() argument
62 if (merge_op == "Add") { in ReductionOp()
65 } else if (merge_op == "Mul") { in ReductionOp()
70 merge_op); in ReductionOp()
76 Status s = ReductionOp(col_params_->merge_op->type_string(), &reduction_op); in Run()
|
| D | collective_nccl_reducer_test.cc | 246 col_params_.merge_op = GetAdd(device_); in DoReduce()
|
| D | collective_ops.cc | 162 col_params_.merge_op = BuildOpKernel(c, merge_op_name, &sub_node); in CollectiveReduceOpKernel()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
56 merge_op=merge_op,
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_lower_alu_to_scalar.c | 43 lower_reduction(nir_alu_instr *instr, nir_op chan_op, nir_op merge_op, in lower_reduction() argument
66 last = nir_build_alu(builder, merge_op, in lower_reduction()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.cc | 405 << ((col_params_->merge_op != nullptr) ? "tmp_chunk" : "chunk"); in DispatchRecv()
406 Tensor* dst_tensor = (!rf->second_pass && (col_params_->merge_op != nullptr)) in DispatchRecv()
|
| D | ring_reducer.cc | 236 col_params_->merge_op.get(), &rf->chunk, &rf->tmp_chunk); in RunAsyncParts()
|
| D | ring_reducer_test.cc | 466 col_params_.merge_op = in DoReduce()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass.cc | 92 ops::Merge merge_op(s.WithOpName(absl::StrCat("merge_oidx_", oidx)), in MergeOutgoingDataEdges() local
94 merged_output = merged_outputs[oidx] = merge_op.output; in MergeOutgoingDataEdges()
|
| /external/v8/src/compiler/ |
| D | simplified-lowering.cc | 3571 const Operator* const merge_op = common()->Merge(2); in Int32Div() local
3603 if_false1 = graph()->NewNode(merge_op, if_true2, if_false2); in Int32Div()
3607 if_false0 = graph()->NewNode(merge_op, if_true1, if_false1); in Int32Div()
3611 Node* merge0 = graph()->NewNode(merge_op, if_true0, if_false0); in Int32Div()
3648 const Operator* const merge_op = common()->Merge(2); in Int32Mod() local
3684 if_false1 = graph()->NewNode(merge_op, if_true2, if_false2); in Int32Mod()
3688 if_true0 = graph()->NewNode(merge_op, if_true1, if_false1); in Int32Mod()
3705 if_false0 = graph()->NewNode(merge_op, if_true1, if_false1); in Int32Mod()
3709 Node* merge0 = graph()->NewNode(merge_op, if_true0, if_false0); in Int32Mod()
3774 const Operator* const merge_op = common()->Merge(2); in Uint32Mod() local
[all …]
|
| D | wasm-compiler.cc | 2306 const Operator* const merge_op = c->Merge(2); in BuildI32AsmjsRemS() local
2340 if_false1 = graph()->NewNode(merge_op, if_true2, if_false2); in BuildI32AsmjsRemS()
2344 if_true0 = graph()->NewNode(merge_op, if_true1, if_false1); in BuildI32AsmjsRemS()
2361 if_false0 = graph()->NewNode(merge_op, if_true1, if_false1); in BuildI32AsmjsRemS()
2365 Node* merge0 = graph()->NewNode(merge_op, if_true0, if_false0); in BuildI32AsmjsRemS()
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | folding_rules.cpp | 762 SpvOp merge_op = inst->opcode(); in MergeDivDivArithmetic() local
765 merge_op = SpvOpFMul; in MergeDivDivArithmetic()
772 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeDivDivArithmetic()
1207 SpvOp merge_op = inst->opcode(); in MergeSubSubArithmetic() local
1209 merge_op = uses_float ? SpvOpFAdd : SpvOpIAdd; in MergeSubSubArithmetic()
1214 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeSubSubArithmetic()
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | folding_rules.cpp | 762 SpvOp merge_op = inst->opcode(); in MergeDivDivArithmetic() local
765 merge_op = SpvOpFMul; in MergeDivDivArithmetic()
772 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeDivDivArithmetic()
1207 SpvOp merge_op = inst->opcode(); in MergeSubSubArithmetic() local
1209 merge_op = uses_float ? SpvOpFAdd : SpvOpIAdd; in MergeSubSubArithmetic()
1214 PerformOperation(const_mgr, merge_op, const_input1, const_input2); in MergeSubSubArithmetic()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 122 std::unique_ptr<OpKernel> merge_op; // reduction only member
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 214 merge_op = control_flow_ops.merge(switch_op)[0]
216 val = merge_op.values
217 ind = merge_op.indices
242 merge_op = control_flow_ops.merge(switch_op)[0]
244 result = self.evaluate(merge_op)
256 merge_op = control_flow_ops.merge([add_op, id_op])[0]
258 result = self.evaluate(merge_op)
271 merge_op = control_flow_ops.merge([add_op, mul_op])[0]
273 result = self.evaluate(merge_op)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 593 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'merge_op\', \'final_…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 593 …argspec: "args=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'merge_op\', \'final_…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4121 …input tf.Output, group_size int64, group_key int64, instance_key int64, merge_op string, final_op …
4125 …size, "group_key": group_key, "instance_key": instance_key, "merge_op": merge_op, "final_op": fina…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 13079 name: "merge_op"
13142 name: "merge_op"
|
| D | ops_history.v1.pbtxt | 13137 name: "merge_op"
13200 name: "merge_op"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 5668 name: "merge_op"
|