Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TpuShapeRepresentation (0.23 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/jit/xla_tpu_device.cc 

    // the TPU device? This function tranposes or flattens the excessively-padded
// tensors to rank 1, but leaves other tensor shapes alone.
absl::StatusOr<xla::Shape> TpuShapeRepresentation(
    const TensorShape& shape, DataType type, bool use_fast_memory,
    XlaLayoutPreference layout_preference) {
  xla::Shape xla_shape;
  TF_RETURN_IF_ERROR(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 22:53:47 UTC 2024
    - 20.9K bytes
    - Viewed (0)
Back to top