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.cc199 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
238 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
Dpartitioning_utils.cc79 std::vector<AllocatorAttributes>* arg_alloc_attrs, in UpdateArgAndRetvalMetadata()
Dprocess_function_library_runtime.h253 std::vector<AllocatorAttributes> arg_alloc_attrs; member
Dfunction.cc1029 AllocatorAttributes arg_alloc_attrs; in RunRemote() local
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc338 AllocatorAttributes arg_alloc_attrs; in ComputeAsync() local