Searched refs:GetInputMemoryTypes (Results 1 – 4 of 4) sorted by relevance
152 GetInputMemoryTypes(fbody, constant_arg_indices, resource_arg_indices); in CreateXlaKernel()
325 tensorflow::MemoryTypeVector GetInputMemoryTypes(
66 GetInputMemoryTypes(fbody, constant_arg_indices, resource_arg_indices); in GetCompilerIr()
656 tensorflow::MemoryTypeVector GetInputMemoryTypes( in GetInputMemoryTypes() function