HomeSort by relevance Sort by last modified time
    Searched refs:ReducePrecisionInsertion (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
reduce_precision_insertion.h 32 class ReducePrecisionInsertion : public HloPassInterface {
40 explicit ReducePrecisionInsertion(
52 explicit ReducePrecisionInsertion(
60 ~ReducePrecisionInsertion() override{};
83 // Add ReducePrecisionInsertion passes to an HloPassPipeline based on the list
reduce_precision_insertion.cc 24 std::vector<HloInstruction*> ReducePrecisionInsertion::instructions_to_modify(
68 StatusOr<bool> ReducePrecisionInsertion::insert_after(
103 StatusOr<bool> ReducePrecisionInsertion::insert_on_inputs(
163 StatusOr<bool> ReducePrecisionInsertion::insert_on_outputs(
196 StatusOr<bool> ReducePrecisionInsertion::Run(HloModule* module) {
198 VLOG(1) << "Running ReducePrecisionInsertion pass on " << module->name();
205 computation_changed = ReducePrecisionInsertion::insert_on_inputs(
212 computation_changed = ReducePrecisionInsertion::insert_on_outputs(
232 ReducePrecisionInsertion::InstructionFilterFunction
233 ReducePrecisionInsertion::make_filter_function
    [all...]
reduce_precision_insertion_test.cc 40 ReducePrecisionInsertion op_insertion(5, 10, location, filter);
393 // The ReducePrecisionInsertion pass should not see inside the fusion
510 auto options_proto = ReducePrecisionInsertion::make_options_proto(
515 ReducePrecisionInsertion::make_filter_function(options_proto);
539 auto options_proto = ReducePrecisionInsertion::make_options_proto(
545 ReducePrecisionInsertion::make_filter_function(options_proto);
  /external/tensorflow/tensorflow/compiler/xla/tests/
reduce_precision_test.cc 249 // ReducePrecisionInsertion passes.
274 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
300 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
324 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
349 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
374 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_compiler.cc 138 ReducePrecisionInsertion::AddPasses(
140 ReducePrecisionInsertion::PassTiming::BEFORE_OPTIMIZATION);
254 ReducePrecisionInsertion::AddPasses(
256 ReducePrecisionInsertion::PassTiming::AFTER_FUSION);
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_compiler.cc 240 ReducePrecisionInsertion::AddPasses(
242 ReducePrecisionInsertion::PassTiming::BEFORE_OPTIMIZATION);
290 ReducePrecisionInsertion::AddPasses(
292 ReducePrecisionInsertion::PassTiming::AFTER_FUSION);
    [all...]

Completed in 223 milliseconds