Searched defs:reduction_type (Results 1 – 7 of 7) sorted by relevance
39 const string& reduction_type) in TypedConditionalAccumulatorBase()
56 const string& name, const string& reduction_type) in ConditionalAccumulator()
23 const string& reduction_type) in ConditionalAccumulatorBase()
53 const string& name, const string& reduction_type) in SparseConditionalAccumulator()
31 DataType reduction_type) in XlaReductionOp()
42 const DataType reduction_type) in PoolingOp()424 auto reduction_type = XlaHelpers::SumAccumulationType(ctx->input_type(1)); in Compile() local
1256 reduction_type="MEAN"): argument1355 reduction_type="MEAN"): argument