Home
last modified time | relevance | path

Searched refs:ComputationMap (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_context.h152 using ComputationMap = std::map<DataType, xla::XlaComputation>; variable
158 DataType type, ComputationMap* out,
162 ComputationMap max_func_;
165 ComputationMap min_func_;
168 ComputationMap add_func_;
171 ComputationMap mul_func_;
174 ComputationMap sigmoid_func_;
Dxla_context.cc150 DataType type, ComputationMap* out, in LookupOrCreate()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcopy_insertion.cc483 ComputationMap; typedef in xla::__anonb2d115a80111::LiveRangeRegions
493 ComputationMap::const_iterator p = computation_map_.find(computation); in operator []()
497 ComputationMap::const_iterator begin() const { in begin()
500 ComputationMap::const_iterator end() const { return computation_map_.end(); } in end()
521 ComputationMap computation_map_;
Dsharding_propagation.cc54 using ComputationMap = typedef
297 const ComputationMap& computation_map, in SupportSpatialPartitioning()
595 const ComputationMap& computation_map, in InferShardingFromOperands()
1427 const ComputationMap& computation_map, in InferShardingFromUsers()
1684 ComputationMap computation_map; in Run()
/external/swiftshader/third_party/subzero/src/
DIceTargetLoweringMIPS32.h884 using ComputationMap = CfgUnorderedMap<SizeT, ComputationEntry>; variable
885 ComputationMap KnownComputations;
DIceTargetLoweringARM32.h1188 using ComputationMap = CfgUnorderedMap<SizeT, ComputationEntry>; variable
1189 ComputationMap KnownComputations;