- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for ShapeDeterminationFn (0.22 sec)
-
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) -
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)