Home
last modified time | relevance | path

Searched defs:accumulation_type (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlrn_ops.cc51 auto accumulation_type = XlaHelpers::SumAccumulationType(input_type(0)); in Compile() local
137 auto accumulation_type = XlaHelpers::SumAccumulationType(input_type(0)); in Compile() local
H A Dfake_quantize_ops.cc198 const DataType accumulation_type = in Compile() local
313 const DataType accumulation_type = in Compile() local
H A Dl2loss_op.cc38 const DataType accumulation_type = XlaHelpers::SumAccumulationType(dtype); in Compile() local
H A Dlower_upper_bound_ops.cc67 const DataType accumulation_type = XlaHelpers::SumAccumulationType(out_dtype); in BuildLowerUpperBoundOp() local
H A Dsoftmax_op.cc66 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in CrossEntropyWithLogits() local
H A Dbatch_norm_op.cc313 const DataType accumulation_type = in Compile() local
H A Dimage_ops.cc204 const DataType accumulation_type = XlaHelpers::SumAccumulationType(type); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dresult_caster_test.cc52 const PrimitiveType accumulation_type = in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtoco_flags.proto251 optional IODataType accumulation_type = 37; field
H A Dargs.h198 Arg<std::string> accumulation_type; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dconvert.py491 accumulation_type=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dpooling.cc81 PrimitiveType accumulation_type = init_shape.element_type(); in ComputeSums() local