Searched defs:accumulation_type (Results 1 – 12 of 12) 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
67 const DataType accumulation_type = XlaHelpers::SumAccumulationType(out_dtype); in BuildLowerUpperBoundOp() local
66 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in CrossEntropyWithLogits() local
198 const DataType accumulation_type = in Compile() local313 const DataType accumulation_type = in Compile() local
313 const DataType accumulation_type = in Compile() local
204 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in Compile() local
52 const PrimitiveType accumulation_type = in TEST_P() local
251 optional IODataType accumulation_type = 37; field
198 Arg<std::string> accumulation_type; member
491 accumulation_type=None, argument
81 PrimitiveType accumulation_type = init_shape.element_type(); in ComputeSums() local