Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_context.h127 using ComputationMap = std::map<DataType, xla::XlaComputation>; variable
133 DataType type, ComputationMap* out,
137 ComputationMap max_func_;
140 ComputationMap min_func_;
143 ComputationMap add_func_;
146 ComputationMap mul_func_;
149 ComputationMap sigmoid_func_;
Dxla_context.cc150 DataType type, ComputationMap* out, in LookupOrCreate()
/external/tensorflow/tensorflow/compiler/xla/service/
Dsharding_propagation.cc52 using ComputationMap = typedef
285 const ComputationMap& computation_map, in SupportSpatialPartitioning()
578 const ComputationMap& computation_map, in InferShardingFromOperands()
1408 const ComputationMap& computation_map, in InferShardingFromUsers()
1667 ComputationMap computation_map; in Run()
/external/swiftshader/third_party/subzero/src/
DIceTargetLoweringMIPS32.h916 using ComputationMap = CfgUnorderedMap<SizeT, ComputationEntry>; variable
917 ComputationMap KnownComputations;
DIceTargetLoweringARM32.h1350 using ComputationMap = CfgUnorderedMap<SizeT, ComputationEntry>; variable
1351 ComputationMap KnownComputations;