Searched defs:GetInitialValue (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 4419 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertMeanOp
4435 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertSumOp
4452 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertMaxOp
4470 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertMinOp
4487 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertProdOp
4502 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertAllOp
4517 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertAnyOp
4606 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertArgMaxOp
4627 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertArgMinOp
6045 static Value GetInitialValue(Type reduce_element_type, Location loc, in GetInitialValue() function in mlir::mhlo::__anon43406a1a0111::ConvertUnsortedSegmentMaxOp
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 568 T GetInitialValue(xla::ReductionKind reduction_kind) { in GetInitialValue() function in __anonf775ca150311::CpuAllReduceRendezvous
|