Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dpartitioning_utils_test.cc290 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
316 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
334 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
363 std::vector<AllocatorAttributes> arg_alloc_attrs; in TEST_F() local
H A Dpartitioning_utils.cc157 std::vector<AllocatorAttributes>* arg_alloc_attrs, in UpdateArgAndRetvalMetadata()
H A Dprocess_function_library_runtime.h286 std::vector<AllocatorAttributes> arg_alloc_attrs; member
H A Dfunction.cc1064 AllocatorAttributes arg_alloc_attrs; in RunRemote() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfunction_ops.cc359 AllocatorAttributes arg_alloc_attrs; in ComputeAsync() local