Searched refs:ComputationMap (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.h | 127 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_;
|
| D | xla_context.cc | 150 DataType type, ComputationMap* out, in LookupOrCreate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | sharding_propagation.cc | 52 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/ |
| D | IceTargetLoweringMIPS32.h | 916 using ComputationMap = CfgUnorderedMap<SizeT, ComputationEntry>; variable
917 ComputationMap KnownComputations;
|
| D | IceTargetLoweringARM32.h | 1350 using ComputationMap = CfgUnorderedMap<SizeT, ComputationEntry>; variable
1351 ComputationMap KnownComputations;
|