Searched refs:scores_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | mnist_gradients_test.cc | 184 TF_Tensor* scores_tensor; in TEST_P() local
185 s = GetValue(outputs[0], &scores_tensor); in TEST_P()
189 memcpy(&result_data[0], TF_TensorData(scores_tensor), in TEST_P()
190 TF_TensorByteSize(scores_tensor)); in TEST_P()
211 TF_DeleteTensor(scores_tensor); in TEST_P()
261 TF_Tensor* scores_tensor; in TEST_P() local
262 s = GetValue(outputs[0], &scores_tensor); in TEST_P()
266 memcpy(&result_data[0], TF_TensorData(scores_tensor), in TEST_P()
267 TF_TensorByteSize(scores_tensor)); in TEST_P()
288 TF_DeleteTensor(scores_tensor); in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | recognize_commands.cc | 79 const Tensor& scores_tensor = previous_result.second; in ProcessLatestResults() local
80 auto scores_flat = scores_tensor.flat<float>(); in ProcessLatestResults()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5924 nvinfer1::ITensor* scores_tensor = inputs.at(1).tensor(); in ConvertCombinedNMS() local
5932 const auto scores_dims = scores_tensor->getDimensions(); in ConvertCombinedNMS()
6056 plugin_inputs.push_back(scores_tensor); in ConvertCombinedNMS()
|
| D | convert_nodes_test.cc | 3136 auto scores_tensor = ops::Placeholder(s.WithOpName("scores"), tf_type); in TEST_P() local
3150 s.WithOpName("my_nms"), boxes_tensor, scores_tensor, in TEST_P()
|