Searched refs:GetInitialValue (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 522 T out = GetInitialValue<T>(reduction_kind); in DoAllReduce()
538 T GetInitialValue(xla::ReductionKind reduction_kind) { in GetInitialValue() function in __anon95669c100311::CpuAllReduceRendezvous
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 3663 Value init = Derived::GetInitialValue(reduce_element_type, loc, &rewriter); in matchAndRewrite()
3714 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon30efc1950111::ConvertMeanOp
3730 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon30efc1950111::ConvertSumOp
3747 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon30efc1950111::ConvertMaxOp
3765 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon30efc1950111::ConvertMinOp
3782 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon30efc1950111::ConvertProdOp
3797 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon30efc1950111::ConvertAllOp
3812 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon30efc1950111::ConvertAnyOp
3842 Derived::GetInitialValue(input_element_type, loc, rewriter); in matchAndRewrite()
3902 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon30efc1950111::ConvertArgMaxOp
[all …]
|