Searched refs:GetOutputMemoryTypes (Results 1 – 5 of 5) sorted by relevance
65 MemoryTypeVector output_memory_types = GetOutputMemoryTypes(fbody); in CreateXlaKernel()
404 StatusOr<MemoryTypeVector> GetOutputMemoryTypes(const Scope& root, Node* n) { in GetOutputMemoryTypes() function436 GetOutputMemoryTypes(root, e->src())); in PredicateInt32Inputs()
330 tensorflow::MemoryTypeVector GetOutputMemoryTypes(
101 MemoryTypeVector output_memory_types = GetOutputMemoryTypes(fbody); in GetCompilerIr()
692 tensorflow::MemoryTypeVector GetOutputMemoryTypes( in GetOutputMemoryTypes() function