Searched defs:reduction_type (Results 1 – 11 of 11) 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()
57 reduction_type=None, argument
31 DataType reduction_type) in XlaReductionOp()
49 const DataType reduction_type) in PoolingOp()493 auto reduction_type = XlaHelpers::SumAccumulationType(ctx->input_type(1)); in Compile() local
59 std::string reduction_type; member
237 auto reduction_type = in CreateNgramsCustomOption() local
1336 reduction_type="MEAN"): argument1438 reduction_type="MEAN"): argument
171 std::optional<ReductionKind> reduction_type = in IsAllReduceMovable() local