Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 52 of 52 for use_cudnn_on_gpu (0.2 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir

        %1 = "tf.Conv2DBackpropInput"(%0, %arg0, %arg1) {data_format = "NHWC", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "VALID", strides = [1, 1, 1, 1], use_cudnn_on_gpu = true} : (tensor<4xi32>, tensor<3x3x32x64xf32>, tensor<200x24x24x64xf32>) -> tensor<?x?x?x?xf32>
        func.return %1 : tensor<?x?x?x?xf32>
      }
    
      // CHECK-LABEL: func @shape_from_if_to_branch_functions_to_results
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 23 17:24:10 UTC 2024
    - 167.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

    `[filter_height, filter_width, in_channels, out_channels]`}]>:$filter,
    
        I64ArrayAttr:$strides,
        DefaultValuedOptionalAttr<BoolAttr, "true">:$use_cudnn_on_gpu,
        TF_AnyStrAttrOf<["SAME", "VALID", "EXPLICIT"]>:$padding,
        DefaultValuedOptionalAttr<I64ArrayAttr, "{}">:$explicit_paddings,
        DefaultValuedOptionalAttr<TF_ConvnetDataFormatAttr, "\"NHWC\"">:$data_format,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top