Searched refs:ResetTensor (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | generate_box_proposals_op.cu.cc | 175 Status ResetTensor(Tensor* t, const Eigen::GpuDevice& d) { in ResetTensor() function
195 TF_RETURN_IF_ERROR(ResetTensor<int>(d_conv_layer_indexes, d)); in AllocateGenerationTempTensors()
198 TF_RETURN_IF_ERROR(ResetTensor<int>(d_image_offset, d)); in AllocateGenerationTempTensors()
205 TF_RETURN_IF_ERROR(ResetTensor<int32>(d_sorted_conv_layer_indexes, d)); in AllocateGenerationTempTensors()
209 TF_RETURN_IF_ERROR(ResetTensor<float>(d_sorted_scores, d)); in AllocateGenerationTempTensors()
213 TF_RETURN_IF_ERROR(ResetTensor<float>(dev_boxes, d)); in AllocateGenerationTempTensors()
217 TF_RETURN_IF_ERROR(ResetTensor<int8>(dev_boxes_keep_flags, d)); in AllocateGenerationTempTensors()
232 TF_RETURN_IF_ERROR(ResetTensor<float>(dev_image_prenms_boxes, d)); in AllocatePreNMSTempTensors()
237 TF_RETURN_IF_ERROR(ResetTensor<float>(dev_image_prenms_scores, d)); in AllocatePreNMSTempTensors()
242 TF_RETURN_IF_ERROR(ResetTensor<int32>(dev_image_boxes_keep_list, d)); in AllocatePreNMSTempTensors()
[all …]
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | tflite_driver_test.cc | 42 runner->ResetTensor("c"); in TEST()
73 runner->ResetTensor("c"); in TEST()
103 runner->ResetTensor("c"); in TEST()
140 runner->ResetTensor(2); in TEST()
|
| D | tf_driver.cc | 162 ResetTensor(name); in ReshapeTensor()
165 void TfDriver::ResetTensor(const std::string& name) { in ResetTensor() function in tflite::testing::TfDriver
|
| D | test_runner_test.cc | 36 void ResetTensor(const std::string& name) override {} in ResetTensor() function in tflite::testing::__anon6fd08a210111::ConcreteTestRunner
|
| D | test_runner.h | 51 virtual void ResetTensor(const std::string& name) = 0;
|
| D | tf_driver.h | 53 void ResetTensor(const std::string& name) override;
|
| D | tf_driver_test.cc | 102 runner->ResetTensor("c"); in TEST()
|
| D | tflite_driver.h | 64 void ResetTensor(const std::string& name) override;
|
| D | parse_testdata.cc | 388 test_runner_->ResetTensor(name); in SetField()
|
| D | tflite_driver.cc | 509 void TfLiteDriver::ResetTensor(const std::string& name) { in ResetTensor() function in tflite::testing::TfLiteDriver
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | tflite_driver.h | 49 void ResetTensor(int id) override;
|
| D | test_runner.h | 56 virtual void ResetTensor(int id) = 0;
|
| D | tflite_driver.cc | 255 void TfLiteDriver::ResetTensor(int id) { in ResetTensor() function in tflite::testing::TfLiteDriver
|
| D | parse_testdata.cc | 332 test_runner_->ResetTensor(id); in SetField()
|