Home
last modified time | relevance | path

Searched defs:AllReduceOp (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc338 TFE_Op* AllReduceOp(TFE_Context* ctx, TFE_TensorHandle* in, int group_size) { in AllReduceOp() function
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dlmhlo_to_jitrt.cc1133 static StringRef CustomCallTarget(AllReduceOp) { in CustomCallTarget() argument
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DStablehloOps.cpp679 INFER_RETURN_TYPE_COMPONENTS_FROM_OPERANDS(AllReduceOp) in INFER_RETURN_TYPE_COMPONENTS_FROM_OPERANDS() argument
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc675 INFER_RETURN_TYPE_COMPONENTS_FROM_OPERANDS(AllReduceOp) in INFER_RETURN_TYPE_COMPONENTS_FROM_OPERANDS() argument