Searched refs:AddObjectRef (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.cc | 162 args_.AddObjectRef(tensor_name, AccessType::READ, std::move(desc_new)); in AddSrcTensor()
169 args_.AddObjectRef(buffer_name, AccessType::READ, std::move(desc_new)); in AddSrcBuffer()
176 args_.AddObjectRef(texture_name, AccessType::READ, std::move(desc_new)); in AddSrcTexture2D()
183 args_.AddObjectRef(tensor_name, AccessType::WRITE, std::move(desc_new)); in AddDstTensor()
194 args_.AddObjectRef("src_tensor", AccessType::READ, std::move(src_desc)); in AssembleCode()
202 args_.AddObjectRef("dst_tensor", AccessType::WRITE, std::move(dst_desc)); in AssembleCode()
|
| D | arguments.h | 62 void AddObjectRef(const std::string& name, AccessType access_type,
|
| D | arguments.cc | 79 void Arguments::AddObjectRef(const std::string& name, AccessType access_type, in AddObjectRef() function in tflite::gpu::Arguments
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_arguments_test.cc | 37 args.AddObjectRef("weights", AccessType::READ, in TEST()
61 args.AddObjectRef("weights", AccessType::READ, in TEST()
|
| D | serialization.cc | 565 args->AddObjectRef(key, access_type, in Decode()
575 args->AddObjectRef(key, access_type, in Decode()
585 args->AddObjectRef( in Decode()
596 args->AddObjectRef(key, access_type, in Decode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | converter.cc | 128 args.AddObjectRef( in Init()
136 args.AddObjectRef( in Init()
217 args.AddObjectRef("tensor", AccessType::READ, in Init()
315 args.AddObjectRef("tensor", AccessType::WRITE, in Init()
|