| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_tensorflow_merge.cc | 32 const auto* merge_op = merge_it->get(); in Run() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_collective.cc | 136 StringRef mode, Value input, StringRef merge_op, in ConvertAllReduce()
215 StringRef merge_op, final_op; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 23 merge_op='Add', argument
99 merge_op='Add', argument
|
| D | collective_ops_gpu_test.py | 250 def run_all_reduce(group_key, instance_key, merge_op): argument
|
| D | collective_ops_test.py | 52 merge_op='Add', argument
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 45 Status ReductionOp(const string& merge_op, ncclRedOp_t* reduction_op) { in ReductionOp()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond_test.cc | 133 auto merge_op = ops::Merge( in TEST() local
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 390 const std::string merge_op("Add"); in CollectiveSum() local
480 const std::string merge_op("Mul"); in RegisterCollectiveMulFunction() local
|
| /external/mesa3d/src/compiler/spirv/ |
| D | vtn_structured_cfg.c | 294 SpvOp merge_op = block->merge[0] & SpvOpCodeMask; in structured_post_order_traversal() local
536 SpvOp merge_op = block->merge[0] & SpvOpCodeMask; in create_constructs() local
1395 SpvOp merge_op = block->merge[0] & SpvOpCodeMask; in vtn_emit_block() local
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_lower_alu_width.c | 96 lower_reduction(nir_alu_instr *alu, nir_op chan_op, nir_op merge_op, in lower_reduction()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 154 OpKernel* merge_op = nullptr; // reduction only member
|
| /external/federated-compute/fcp/protos/ |
| D | plan.proto | 1357 string merge_op = 3; field
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | folding_rules.cpp | 917 spv::Op merge_op = inst->opcode(); in MergeDivDivArithmetic() local
1390 spv::Op merge_op = inst->opcode(); in MergeSubSubArithmetic() local
|
| /external/angle/third_party/spirv-tools/src/source/opt/ |
| D | folding_rules.cpp | 920 spv::Op merge_op = inst->opcode(); in MergeDivDivArithmetic() local
1393 spv::Op merge_op = inst->opcode(); in MergeSubSubArithmetic() local
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | folding_rules.cpp | 869 spv::Op merge_op = inst->opcode(); in MergeDivDivArithmetic() local
1312 spv::Op merge_op = inst->opcode(); in MergeSubSubArithmetic() local
|