Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ShapeDeterminationFn (0.18 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)
Back to top