- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for use_kernel (0.32 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
'has_bias': [True, False], 'use_kernel': [True, False], }]) ) @test_util.run_in_graph_and_eager_modes def test_qat_matmul_model( self, shapes: Sequence[Tuple[_TensorShape, _TensorShape]], activation_fn: Optional[ops.Operation], has_bias: bool, use_kernel: bool, ): n = 5
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/stablehlo/transforms/legalize_hlo.cc
"multiplier"); } // Slicing with dynamic offsets (helper method advised) auto create_slice = [&](mlir::Value tensor, int depth_idx, int channel_idx, bool is_kernel = false) -> mlir::Value { std::vector<int64_t> tensor_shape = mlir::cast<ShapedType>(tensor.getType()).getShape().vec(); // Calculate offsets based on depth_idx, channel_idx and tensor_shape
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)