Searched refs:SumAccumulationType (Results 1 – 12 of 12) sorted by relevance
34 XlaHelpers::SumAccumulationType(ctx->input_type(0))) {} in SumOp()51 XlaHelpers::SumAccumulationType(ctx->input_type(0))) {} in ProdOp()106 XlaHelpers::SumAccumulationType(ctx->input_type(0))) {} in MeanOp()
51 auto accumulation_type = XlaHelpers::SumAccumulationType(input_type(0)); in Compile()137 auto accumulation_type = XlaHelpers::SumAccumulationType(input_type(0)); in Compile()
38 const DataType accumulation_type = XlaHelpers::SumAccumulationType(dtype); in Compile()
65 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in Compile()112 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in CrossEntropyWithLogits()
112 XlaHelpers::SumAccumulationType(input_type(0)); in Compile()
70 DataType dtype = XlaHelpers::SumAccumulationType(ctx->input_type(0)); in Compile()
184 XlaHelpers::SumAccumulationType(scale_dtype); in Compile()
207 XlaHelpers::SumAccumulationType(ctx->input_type(0))) { in AvgPoolOp()424 auto reduction_type = XlaHelpers::SumAccumulationType(ctx->input_type(1)); in Compile()
244 XlaHelpers::SumAccumulationType(data_type); in Compile()
195 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in Compile()
69 static DataType SumAccumulationType(const DataType& dtype);
108 DataType XlaHelpers::SumAccumulationType(const DataType& dtype) { in SumAccumulationType() function in tensorflow::XlaHelpers