Searched refs:conversion_map_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 208 std::map<TensorDesc, WrappedTensor> conversion_map_; member in tensorflow::grappler::vectorization_utils::__anon1f3551c50111::Vectorization
256 if (auto found = gtl::FindOrNull(conversion_map_, in AddConversionMapping()
291 conversion_map_.insert({{op_node, i}, outputs[i]}); in AddConversionMapping()
309 auto found = gtl::FindOrNull(conversion_map_, output); in ConvertOutput()
312 found = &conversion_map_.at(output); in ConvertOutput()
515 conversion_map_.insert({{arg_node, 0}, {input_node, 0, stacked}}); in AddArgTensorMappings()
518 conversion_map_.insert({{arg_node, Graph::kControlSlot}, in AddArgTensorMappings()
540 if (auto found = gtl::FindOrNull(conversion_map_, tensor)) { in AddUnstackedTensorMappingsHelper()
586 if (auto found = gtl::FindOrNull(conversion_map_, in AddUnstackedTensorMappingsHelper()
597 conversion_map_.insert({{tensor.first, i}, WrappedTensor(node, i, false)}); in AddUnstackedTensorMappingsHelper()
[all …]
|