| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_tensorflow_merge.cc | 32 const auto* merge_op = merge_it->get(); local
33 if (merge_op->type != OperatorType::kMerge) {
41 if (merge_op->inputs.size() > 1) {
42 AddMessageF("Waiting for %s to be resolved", LogName(*merge_op));
48 CHECK_EQ(merge_op->inputs.size(), 1);
53 if (input == merge_op->outputs[0]) {
54 input = merge_op->inputs[0];
60 AddMessageF("Removing already-resolved %s", LogName(*merge_op));
61 model->EraseArray(merge_op->outputs[0]);
|
| /external/tensorflow/tensorflow/python/ops/ |
| collective_ops.py | 24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op,
34 merge_op: string naming the binary Op to be applied to compute each
56 merge_op=merge_op,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| collective_nccl_reducer.cc | 61 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) {
62 if (merge_op == "Add") {
65 } else if (merge_op == "Mul") {
69 return errors::Internal("Expected merge_op to be either Add or Mul, found ",
70 merge_op);
76 Status s = ReductionOp(col_params_->merge_op->type_string(), &reduction_op);
|
| collective_nccl_reducer_test.cc | 204 .Attr("merge_op", "Add")
246 col_params_.merge_op = GetAdd(device_);
|
| collective_ops.cc | 135 OP_REQUIRES_OK(c, c->GetAttr("merge_op", &merge_op_name));
138 "merge_op must be one of {\"Add\", \"Mul\"} but got ",
156 // The merge_op takes two inputs
162 col_params_.merge_op = BuildOpKernel(c, merge_op_name, &sub_node);
|
| /external/mesa3d/src/compiler/nir/ |
| nir_lower_alu_to_scalar.c | 43 lower_reduction(nir_alu_instr *instr, nir_op chan_op, nir_op merge_op,
66 last = nir_build_alu(builder, merge_op,
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| build_xla_ops_pass.cc | 92 ops::Merge merge_op(s.WithOpName(absl::StrCat("merge_oidx_", oidx)),
94 merged_output = merged_outputs[oidx] = merge_op.output;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| ring_alg.cc | 405 << ((col_params_->merge_op != nullptr) ? "tmp_chunk" : "chunk");
406 Tensor* dst_tensor = (!rf->second_pass && (col_params_->merge_op != nullptr))
|
| ring_reducer.cc | 236 col_params_->merge_op.get(), &rf->chunk, &rf->tmp_chunk);
|
| ring_reducer_test.cc | 373 .Attr("merge_op", "Add")
466 col_params_.merge_op =
|
| /external/tensorflow/tensorflow/core/framework/ |
| collective.h | 122 std::unique_ptr<OpKernel> merge_op; // reduction only member in struct:tensorflow::CollectiveParams
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| folding_rules.cpp | 762 SpvOp merge_op = inst->opcode();
765 merge_op = SpvOpFMul;
772 PerformOperation(const_mgr, merge_op, const_input1, const_input2);
[all...] |
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| folding_rules.cpp | 762 SpvOp merge_op = inst->opcode();
765 merge_op = SpvOpFMul;
772 PerformOperation(const_mgr, merge_op, const_input1, const_input2);
[all...] |
| /external/v8/src/compiler/ |
| simplified-lowering.cc | 3571 const Operator* const merge_op = common()->Merge(2); local
3648 const Operator* const merge_op = common()->Merge(2); local
3774 const Operator* const merge_op = common()->Merge(2); local
[all...] |
| wasm-compiler.cc | 2306 const Operator* const merge_op = c->Merge(2); local
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| 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)
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
