- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for getUseCudnnOnGpu (0.3 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto loc = backprop.getLoc(); auto new_backprop = builder.create<TF::Conv2DBackpropFilterOp>( loc, new_result_type, input, new_filter_sizes, backprop.getOutBackprop(), strides, backprop.getUseCudnnOnGpu(), backprop.getPadding(), backprop.getExplicitPaddings(), backprop.getDataFormat(), backprop.getDilations()); // For example, if new filter shape is [4, 4, 12, 64], old filter shape
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0)