Searched defs:accumulation_type (Results 1 – 8 of 8) sorted by relevance
51 auto accumulation_type = XlaHelpers::SumAccumulationType(input_type(0)); in Compile() local137 auto accumulation_type = XlaHelpers::SumAccumulationType(input_type(0)); in Compile() local
38 const DataType accumulation_type = XlaHelpers::SumAccumulationType(dtype); in Compile() local
65 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in Compile() local112 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in CrossEntropyWithLogits() local
111 const DataType accumulation_type = in Compile() local
183 const DataType accumulation_type = in Compile() local
243 const DataType accumulation_type = in Compile() local
195 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in Compile() local
81 PrimitiveType accumulation_type = init_shape.element_type(); in ComputeSums() local