Searched refs:ComputationMap (Results 1 – 4 of 4) sorted by relevance
114 using ComputationMap = std::map<DataType, xla::XlaComputation>; variable120 DataType type, ComputationMap* out,124 ComputationMap max_func_;127 ComputationMap min_func_;130 ComputationMap add_func_;133 ComputationMap mul_func_;136 ComputationMap sigmoid_func_;
143 DataType type, ComputationMap* out, in LookupOrCreate()
922 using ComputationMap = CfgUnorderedMap<SizeT, ComputationEntry>; variable923 ComputationMap KnownComputations;
1357 using ComputationMap = CfgUnorderedMap<SizeT, ComputationEntry>; variable1358 ComputationMap KnownComputations;