- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetOutputMemoryTypes (0.65 sec)
-
tensorflow/compiler/jit/xla_kernel_creator.cc
MemoryTypeVector input_memory_types = GetInputMemoryTypes(fbody, constant_arg_indices, resource_arg_indices); MemoryTypeVector output_memory_types = GetOutputMemoryTypes(fbody); // Create the kernel. Device* dev = flr->device(); Status s; auto props = std::make_shared<NodeProperties>( &fbody->record->fdef().signature(), node_def, fbody->arg_types,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K bytes - Viewed (0)