Home
last modified time | relevance | path

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

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