- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 76 for conv_2d (0.14 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_custom_aggregation_ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 32.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
// Input: [N, H, W, C] for Conv2D or [N, D, H, W, C] for Conv3D. dnums.set_input_batch_dimension(0); dnums.set_input_feature_dimension(num_dims - 1); // Kernel: [K, K, I, O] for Conv2D or [K, K, K, I, O] for Conv3D. dnums.set_kernel_input_feature_dimension(num_dims - 2); dnums.set_kernel_output_feature_dimension(num_dims - 1); // Output: [N, H, W, C] for Conv2D or [N, D, H, W, C] for Conv3D.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_patterns.td
(UpdateShapeWithAxis<-1> $qtype, $old_value))), [(CanUpdateShapeWithAxis<-1> $qtype, $old_value)]>; // The axis is set to 0 because the transpose is from the legalization of // tf.conv2d and the new channel axis is the first dimension. def ReorderTransposeDequantQuantUsedByConv : Pat<(TF_TransposeOp:$old_value (TFL_DequantizeOp (TFL_QuantizeOp $input, $qtype)), $perm),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/arithmetic_count_util.h
if (!input_type || !input_type.hasStaticShape()) { return false; } total_count += input_type.getNumElements(); } *count = total_count; return true; } // For conv2d/depthwise_conv/fully_connected ops. // This algorithm actually comes from TOCO tooling_util.cc static bool GetArithmeticCountForConvAndFullyconnectedOp(mlir::Operation* op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_xla.mlir
%dq_bias = "quantfork.dcast"(%q_bias) : (tensor<2x!quant.uniform<i32:f32, 0.044022349891595126>>) -> tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver_with_skipping.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize.mlir
%dq_bias = "quantfork.dcast"(%q_bias) : (tensor<2x!quant.uniform<i32:f32, 0.044022349891595126>>) -> tensor<2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/tf_to_corert_pipeline.mlir
%outputs_6, %control_7 = tf_executor.island wraps "tf.Const"() {device = "", value = dense<[-1, 16384]> : tensor<2xi32>} : () -> tensor<2xi32> %outputs_8, %control_9 = tf_executor.island wraps "tf.Conv2D"(%arg0, %outputs_0) {data_format = "NHWC", device = "", dilations = [1, 1, 1, 1], explicit_paddings = [], padding = "SAME", strides = [1, 2, 2, 1], use_cudnn_on_gpu = true} : (tensor<16x224x224x3xf32>, tensor<*xf32>) -> tensor<16x112x112x?xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.cc
// Canonicalization includes const folding, which is utilized here to optimize // away ops that can't get constant folded after PrepareTF pass. For example, // tf.Conv2D is split into tf.Transpose and tfl.Conv2D. pass_manager->addNestedPass<mlir::func::FuncOp>( mlir::createCanonicalizerPass()); pass_manager->addNestedPass<mlir::func::FuncOp>(mlir::createCSEPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 25.5K bytes - Viewed (0)