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
310 const DataType accumulation_type = in Compile() local
242 const DataType accumulation_type = in Compile() local
197 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
196 Arg<std::string> accumulation_type; member
448 accumulation_type=None, argument496 def build_toco_convert_protos(input_tensors,
81 PrimitiveType accumulation_type = init_shape.element_type(); in ComputeSums() local