Home
last modified time | relevance | path

Searched defs:merge_op (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_tensorflow_merge.cc32 const auto* merge_op = merge_it->get(); in Run() local
/external/tensorflow/tensorflow/python/ops/
Dcollective_ops.py24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, argument
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc61 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp()
/external/mesa3d/src/compiler/nir/
Dnir_lower_alu_to_scalar.c43 lower_reduction(nir_alu_instr *instr, nir_op chan_op, nir_op merge_op, in lower_reduction()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass.cc92 ops::Merge merge_op(s.WithOpName(absl::StrCat("merge_oidx_", oidx)), in MergeOutgoingDataEdges() local
/external/tensorflow/tensorflow/core/framework/
Dcollective.h122 std::unique_ptr<OpKernel> merge_op; // reduction only member
/external/v8/src/compiler/
Dsimplified-lowering.cc3571 const Operator* const merge_op = common()->Merge(2); in Int32Div() local
3648 const Operator* const merge_op = common()->Merge(2); in Int32Mod() local
3774 const Operator* const merge_op = common()->Merge(2); in Uint32Mod() local
Dwasm-compiler.cc2306 const Operator* const merge_op = c->Merge(2); in BuildI32AsmjsRemS() local
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dfolding_rules.cpp762 SpvOp merge_op = inst->opcode(); in MergeDivDivArithmetic() local
1207 SpvOp merge_op = inst->opcode(); in MergeSubSubArithmetic() local
/external/deqp-deps/SPIRV-Tools/source/opt/
Dfolding_rules.cpp762 SpvOp merge_op = inst->opcode(); in MergeDivDivArithmetic() local
1207 SpvOp merge_op = inst->opcode(); in MergeSubSubArithmetic() local