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

  /external/tensorflow/tensorflow/compiler/xla/service/
reduce_precision_insertion.h 31 class ReducePrecisionInsertion : public HloModulePass {
39 explicit ReducePrecisionInsertion(
51 explicit ReducePrecisionInsertion(
59 ~ReducePrecisionInsertion() override{};
82 // 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);
420 // The ReducePrecisionInsertion pass should not see inside the fusion
537 auto options_proto = ReducePrecisionInsertion::make_options_proto(
542 ReducePrecisionInsertion::make_filter_function(options_proto);
566 auto options_proto = ReducePrecisionInsertion::make_options_proto(
572 ReducePrecisionInsertion::make_filter_function(options_proto);
  /external/tensorflow/tensorflow/compiler/xla/tests/
reduce_precision_test.cc 247 // ReducePrecisionInsertion passes.
272 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
298 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
322 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
347 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
372 *reduce_precision_pass = ReducePrecisionInsertion::make_options_proto(
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
compiler.cc 90 ReducePrecisionInsertion::AddPasses(
92 ReducePrecisionInsertion::PassTiming::BEFORE_OPTIMIZATION);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
nvptx_compiler.cc 188 ReducePrecisionInsertion::AddPasses(
190 ReducePrecisionInsertion::PassTiming::BEFORE_OPTIMIZATION);
380 ReducePrecisionInsertion::AddPasses(
382 ReducePrecisionInsertion::PassTiming::AFTER_FUSION);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_compiler.cc 256 ReducePrecisionInsertion::AddPasses(
258 ReducePrecisionInsertion::PassTiming::BEFORE_OPTIMIZATION);
331 ReducePrecisionInsertion::AddPasses(
333 ReducePrecisionInsertion::PassTiming::AFTER_FUSION);
    [all...]

Completed in 2675 milliseconds