Home
last modified time | relevance | path

Searched refs:GetOutputMemoryTypes (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_kernel_creator.cc65 MemoryTypeVector output_memory_types = GetOutputMemoryTypes(fbody); in CreateXlaKernel()
H A Dbuild_xla_ops_pass.cc404 StatusOr<MemoryTypeVector> GetOutputMemoryTypes(const Scope& root, Node* n) { in GetOutputMemoryTypes() function
436 GetOutputMemoryTypes(root, e->src())); in PredicateInt32Inputs()
H A Dcompilability_check_util.h330 tensorflow::MemoryTypeVector GetOutputMemoryTypes(
H A Dget_compiler_ir.cc101 MemoryTypeVector output_memory_types = GetOutputMemoryTypes(fbody); in GetCompilerIr()
H A Dcompilability_check_util.cc692 tensorflow::MemoryTypeVector GetOutputMemoryTypes( in GetOutputMemoryTypes() function