- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for XlaShapeToTpuPaddedShapeFn (0.2 sec)
-
tensorflow/compiler/jit/xla_tpu_device.cc
StatusHelper status; ApiConverter::StackHelper<XLA_Shape> se_shape(on_device_shape); ApiConverter::StackHelper<XLA_Shape> tpu_shape; stream_executor::tpu::ExecutorApiFn()->XlaShapeToTpuPaddedShapeFn( &se_shape.value, &tpu_shape.value, status.c_status); if (!status.ok()) { return status.status(); } *shape = tpu_shape.AsCpp<xla::Shape>(); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0)