- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetInputMemoryTypes (0.34 sec)
-
tensorflow/compiler/jit/xla_kernel_creator.cc
std::vector<int> resource_arg_indices; TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources( flr, function, &fbody, &constant_arg_indices, &resource_arg_indices)); 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;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 22:24:01 UTC 2023 - 4.4K bytes - Viewed (0)