- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 44 for 2xi64 (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/convert_session_initializer_to_function.mlir
%0 = "tf.Const"() {value = dense<42> : tensor<1xi64>} : () -> tensor<1xi64> return } func.func @init2() attributes {tf_saved_model.exported_names = ["init2"]} { %0 = "tf.Const"() {value = dense<43> : tensor<1xi64>} : () -> tensor<1xi64> return } func.func @init3() attributes {tf_saved_model.exported_names = ["init3"]} { %0 = "tf.Const"() {value = dense<44> : tensor<1xi64>} : () -> tensor<1xi64> return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 02 16:24:00 UTC 2022 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/layout_optimization_layout_assignment_to_nchw.mlir
// CHECK-LABEL: func @transposeConv2D func.func @transposeConv2D(%input: tensor<1x32x32x3xf32>, %filter: tensor<1x1x3x8xf32>) -> tensor<1x7x7x8xf32> { // CHECK: %[[ARG_PERM:.*]] = "tf.Const"() <{value = dense<[0, 3, 1, 2]> : tensor<4xi64>}> // CHECK: %[[ARG_TRANSPOSE:[0-9]*]] = "tf.Transpose"(%arg0, %[[ARG_PERM]]) // CHECK: %[[CONV2D:[0-9]*]] = "tf.Conv2D"(%[[ARG_TRANSPOSE]], %arg1) // CHECK-SAME: data_format = "NCHW"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/optimize.mlir
func.return %2, %3 : tensor<*xi64>, tensor<*xi64> } // ----- // CHECK-LABEL: @not_eliminate_multinomial func.func @not_eliminate_multinomial(%0: tensor<*xf32>, %1: tensor<*xi32>) -> (tensor<*xi64>, tensor<*xi64>) { // CHECK-NEXT: tf.Multinomial // CHECK-SAME: seed = 0 // CHECK-NEXT: tf.Multinomial // CHECK-SAME: seed = 1 // CHECK-NEXT: tf.Multinomial
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jul 01 23:50:06 UTC 2023 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/unfold_splat_constant_pass.mlir
func.return %cst : tensor<1x750xf32> // CHECK-DAG: %0 = mhlo.constant dense<7.680000e+02> : tensor<f32> // CHECK: %1 = "mhlo.broadcast_in_dim"(%0) <{broadcast_dimensions = dense<> : tensor<0xi64>}> : (tensor<f32>) -> tensor<1x750xf32> // CHECK: return %1 : tensor<1x750xf32> } // CHECK-LABEL: @unfold_splat_constant_integer func.func @unfold_splat_constant_integer() -> tensor<1x750xi32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize-tfl-stablehlo-constant.mlir
module { func.func @main() -> tensor<1xi64> { %0 = "tfl.custom"() {custom_code = "stablehlo.constant", custom_option = #tfl<const_bytes : "0x76616C75650001020109010101062C022401">} : () -> tensor<1xi64> func.return %0 : tensor<1xi64> } } // CHECK: module { // CHECK-NEXT: func @main() -> tensor<1xi64> { // CHECK-NEXT: %[[c0:.+]] = stablehlo.constant dense<2> : tensor<1xi64> // CHECK-NEXT: return %[[c0]] : tensor<1xi64>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 18:33:43 UTC 2024 - 574 bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/unfuse_mhlo_batch_norm.mlir
// CHECK-DAG: %[[SCALE_BCAST:.+]] = "mhlo.broadcast_in_dim"(%[[SCALE]]) <{broadcast_dimensions = dense<1> : tensor<1xi64>}> : (tensor<256xf32>) -> tensor<4x256xf32> // CHECK-DAG: %[[OFFSET_BCAST:.+]] = "mhlo.broadcast_in_dim"(%[[OFFSET]]) <{broadcast_dimensions = dense<1> : tensor<1xi64>}> : (tensor<256xf32>) -> tensor<4x256xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/components/pre_calibration_component.mlir
// CHECK: %[[XLA_CALL_MODULE:.+]] = "tf.XlaCallModule"(%[[CUSTOM_AGGREGATOR_0]], %[[CST]]) // CHECK-SAME: _entry_function = @composite_dot_general_fn_1, _original_entry_function = "composite_dot_general_fn_1"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/tests/shlo_simplify.mlir
// ----- func.func @foldDivF64() -> tensor<2xf64> { %0 = stablehlo.constant dense<[2.0, 3.0]> : tensor<2xf64> %1 = stablehlo.constant dense<[4.0, 6.0]> : tensor<2xf64> %2 = stablehlo.divide %0, %1 : tensor<2xf64> return %2 : tensor<2xf64> } // CHECK-LABEL: foldDivF64 // CHECK: stablehlo.constant dense<5.000000e-01> : tensor<2xf64> // ----- func.func @foldDivI32() -> tensor<2xi32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 03:05:20 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/verify_tfxla_legalization_test.cc
%0 = mhlo.constant dense<1.000000e+00> : tensor<f64> %1 = mhlo.convert %0 : (tensor<f64>) -> tensor<i64> %2 = mhlo.reshape %1 : (tensor<i64>) -> tensor<1xi64> %3 = "mhlo.dynamic_iota"(%2) {iota_dimension = 0 : i64} : (tensor<1xi64>) -> tensor<?xi32> %4 = mhlo.multiply %3, %3 : tensor<?xi32> return %4 : tensor<?xi32> } })"; CellReader<int64_t> legal_error(kFailedLegalizationStreamz);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/convert_custom_aggregation_op_to_quant_stats.mlir
%1:4 = "tf.CustomAggregator"(%arg0) {id = "1", calibration_method = 1 : i32, num_bins = 0 : i32, max_percentile = 0.000000e+00 : f32, min_percentile = 0.000000e+00 : f32} : (tensor<8x8x8x8xf32>) -> (tensor<8x8x8x8xf32>, tensor<f32>, tensor<f32>, tensor<*xi64>) func.return %0#0, %1#0 : tensor<8x8x8x8xf32>, tensor<8x8x8x8xf32> } // CHECK: func @customAggregator
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 06:25:50 UTC 2024 - 2.2K bytes - Viewed (0)