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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_module.cc 308 tensorflow::gtl::FlatMap<string, HloComputation*> computation_map; local
313 module.get(), computation_proto, computation_map,
321 TF_RET_CHECK(!ContainsKey(computation_map, computation->name()));
325 computation_map[computation_name] = module->AddComputationInternal(
hlo_computation.h 163 // computation_map: a map from computation name to HloComputation*. This map
173 const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
hlo_computation.cc 408 const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
419 computation_map, add_fused_computation));
hlo_instruction.cc 55 const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
86 computation_map, add_fused_computation,
92 TF_RET_CHECK(ContainsKey(computation_map, computation_name))
95 computation_map.at(computation_name));
    [all...]
hlo_instruction.h 184 // computation_map: a map from computation name to HloComputation*. This map
193 const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
    [all...]

Completed in 206 milliseconds