Home
last modified time | relevance | path

Searched defs:arg_alloc_attrs (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils_test.cc202 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
241 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
Dpartitioning_utils.cc76 std::vector<AllocatorAttributes>* arg_alloc_attrs, in UpdateArgAndRetvalMetadata()
Dprocess_function_library_runtime.h265 std::vector<AllocatorAttributes> arg_alloc_attrs; member
Dfunction.cc1123 AllocatorAttributes arg_alloc_attrs; in RunRemote() local
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc344 AllocatorAttributes arg_alloc_attrs; in ComputeAsync() local