Home
last modified time | relevance | path

Searched defs:tf_allocator (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dxsmm_conv2d_test.cc188 const libxsmm_tf_allocator<libxsmm_scratch_allocator> tf_allocator; in TEST_F() local
H A Dsparse_matmul_op.cc910 const libxsmm_tf_allocator<libxsmm_scratch_allocator> tf_allocator; in take_cache_entry() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc246 tensorflow::Allocator* tf_allocator = in InjectTfGpuResourcesHelper() local