OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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