Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GetOutputMemoryTypes (0.65 sec)

  1. 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)
Back to top