Searched refs:InitializeTensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 54 void InitializeTensor(DataType type, Tensor* tensor) { in InitializeTensor() function
239 InitializeTensor(input.dtype(), &value_tensor); in GrapplerItemFromMetaGraphDef()
247 InitializeTensor(input.dtype(), &indices_tensor); in GrapplerItemFromMetaGraphDef()
256 InitializeTensor(input.dtype(), &dense_shape_tensor); in GrapplerItemFromMetaGraphDef()
275 InitializeTensor(input.dtype(), &fake_input); in GrapplerItemFromMetaGraphDef()
497 InitializeTensor(type, &fake_input); in GrapplerItemFromMetaGraphDef()
|
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 63 void InitializeTensor(const std::vector<float>& initialization_values, in InitializeTensor() function
81 InitializeTensor<int32>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
85 InitializeTensor<float>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
89 InitializeTensor<quint8>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
93 InitializeTensor<uint8>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
97 InitializeTensor<bool>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
|