Home
last modified time | relevance | path

Searched refs:reduce_precision_options (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_precision_insertion.h52 const HloReducePrecisionOptions& reduce_precision_options) in ReducePrecisionInsertion() argument
53 : exponent_bits_(reduce_precision_options.exponent_bits()), in ReducePrecisionInsertion()
54 mantissa_bits_(reduce_precision_options.mantissa_bits()), in ReducePrecisionInsertion()
55 location_(reduce_precision_options.location()), in ReducePrecisionInsertion()
57 make_filter_function(reduce_precision_options)) {} in ReducePrecisionInsertion()
72 const HloReducePrecisionOptions& reduce_precision_options);
Dreduce_precision_insertion.cc234 const HloReducePrecisionOptions& reduce_precision_options) { in make_filter_function() argument
237 for (const auto& opcode : reduce_precision_options.opcodes_to_suffix()) { in make_filter_function()
240 if (reduce_precision_options.opname_substrings_to_suffix_size() == 0) { in make_filter_function()
247 reduce_precision_options.opname_substrings_to_suffix()) { in make_filter_function()