Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dlmhlo_to_jitrt.cc1215 static LogicalResult SetSpecificAttrs(ImplicitLocOpBuilder& b, ReduceOp op, in SetSpecificAttrs() function in tensorflow::CollectiveOpLowering
1229 static LogicalResult SetSpecificAttrs(ImplicitLocOpBuilder& b, AllGatherOp op, in SetSpecificAttrs() function in tensorflow::CollectiveOpLowering
1233 static LogicalResult SetSpecificAttrs(ImplicitLocOpBuilder& b, AllToAllOp op, in SetSpecificAttrs() function in tensorflow::CollectiveOpLowering
1239 static LogicalResult SetSpecificAttrs(ImplicitLocOpBuilder& b, in SetSpecificAttrs() function in tensorflow::CollectiveOpLowering