Searched refs:SumAccumulationType (Results 1 – 13 of 13) sorted by relevance
34 XlaHelpers::SumAccumulationType(ctx->input_type(0))) {} in SumOp()51 XlaHelpers::SumAccumulationType(ctx->input_type(0))) {} in ProdOp()120 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()
67 const DataType accumulation_type = XlaHelpers::SumAccumulationType(out_dtype); in BuildLowerUpperBoundOp()
90 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in Compile()137 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in CrossEntropyWithLogits()
112 XlaHelpers::SumAccumulationType(input_type(0)); in Compile()
68 DataType dtype = XlaHelpers::SumAccumulationType(ctx->input_type(0)); in Compile()
215 XlaHelpers::SumAccumulationType(ctx->input_type(0))) { in AvgPoolOp()445 auto reduction_type = XlaHelpers::SumAccumulationType(ctx->input_type(1)); in Compile()
310 XlaHelpers::SumAccumulationType(scale_dtype); in Compile()
295 XlaHelpers::SumAccumulationType(data_type); in Compile()
197 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in Compile()
70 static DataType SumAccumulationType(const DataType& dtype);
106 DataType XlaHelpers::SumAccumulationType(const DataType& dtype) { in SumAccumulationType() function in tensorflow::XlaHelpers