Searched refs:arg_alloc_attrs (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | partitioning_utils_test.cc | 290 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
294 graph.get(), &arg_indices, &ret_indices, &arg_alloc_attrs, in TEST_F()
300 CheckAlloc({false}, arg_alloc_attrs); in TEST_F()
316 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
320 graph.get(), &arg_indices, &ret_indices, &arg_alloc_attrs, in TEST_F()
324 CheckAlloc({true}, arg_alloc_attrs); in TEST_F()
334 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
338 graph.get(), &arg_indices, &ret_indices, &arg_alloc_attrs, in TEST_F()
342 CheckAlloc({false}, arg_alloc_attrs); in TEST_F()
363 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
[all …]
|
| H A D | partitioning_utils.cc | 157 std::vector<AllocatorAttributes>* arg_alloc_attrs, in UpdateArgAndRetvalMetadata() argument
205 if (arg_alloc_attrs != nullptr) { in UpdateArgAndRetvalMetadata()
213 arg_alloc_attrs->push_back(alloc_attr); in UpdateArgAndRetvalMetadata()
|
| H A D | partitioning_utils.h | 79 std::vector<AllocatorAttributes>* arg_alloc_attrs,
|
| H A D | function.cc | 1064 AllocatorAttributes arg_alloc_attrs; in RunRemote() local
1066 arg_alloc_attrs.set_on_host(true); in RunRemote()
1068 args_alloc_attrs.push_back(arg_alloc_attrs); in RunRemote()
|
| H A D | process_function_library_runtime.h | 286 std::vector<AllocatorAttributes> arg_alloc_attrs; member
|
| H A D | process_function_library_runtime.cc | 1077 &comp_data->arg_alloc_attrs, &comp_data->ret_alloc_attrs, in InstantiateMultiDevice()
1295 opts_copy.args_alloc_attrs = comp_data.arg_alloc_attrs; in RunMultiDeviceSync()
1390 opts_copy.args_alloc_attrs = comp_data.arg_alloc_attrs; in RunMultiDeviceAsync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | function_ops.cc | 359 AllocatorAttributes arg_alloc_attrs; in ComputeAsync() local
360 arg_alloc_attrs.set_on_host(DataTypeAlwaysOnHost(dtype)); in ComputeAsync()
361 opts.args_alloc_attrs.push_back(arg_alloc_attrs); in ComputeAsync()
|