Searched refs:InitializeTensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/tools/benchmark/ |
| D | benchmark_model.cc | 64 void InitializeTensor(const std::vector<float>& initialization_values, in InitializeTensor() function
82 InitializeTensor<int32>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
86 InitializeTensor<int64>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
90 InitializeTensor<float>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
94 InitializeTensor<quint8>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
98 InitializeTensor<uint8>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
102 InitializeTensor<bool>(input.initialization_values, &input_tensor); in CreateTensorsFromInputInfo()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 55 void InitializeTensor(DataType type, Tensor* tensor) { in InitializeTensor() function
176 InitializeTensor(type, &fake_input); in UpdatePlaceholderShape()
340 InitializeTensor(input.dtype(), &value_tensor); in GrapplerItemFromMetaGraphDef()
348 InitializeTensor(input.dtype(), &indices_tensor); in GrapplerItemFromMetaGraphDef()
357 InitializeTensor(input.dtype(), &dense_shape_tensor); in GrapplerItemFromMetaGraphDef()
376 InitializeTensor(input.dtype(), &fake_input); in GrapplerItemFromMetaGraphDef()
|