Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for ShapeDeterminationFn (0.35 sec)

  1. tensorflow/compiler/jit/pjrt_base_device.h

        int device_ordinal = -1;
    
        // The name of the compilation device, also referred to as jit_device_type.
        // (e.g., "XLA_CPU_JIT");
        std::string compilation_device_name;
    
        // A vector of ShapeDeterminationFn (i.e., a bundle of LayoutSelectionFn,
        // ShapeRepresentationFn). Each bundle describes how the on-host shapes of
        // a) argument and return value, for entry computations b) variables, for
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 4K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_device.h

        // compute stream. Otherwise each XLA device will having their own compute
        // streams.
        bool use_global_compute_stream = false;
    
        // A vector of ShapeDeterminationFn (i.e., a bundle of LayoutSelectionFn,
        // ShapeRepresentationFn). Each bundle describes how the on-host shapes of
        // a) argument and return value, for entry computations b) variables, for
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 13.4K bytes
    - Viewed (0)
Back to top