- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 166 for mat_mul (0.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.h
namespace mlir { namespace TF { // Populate patterns to unroll tf.BatchMatMulV2 op into a sequence of TF ops. // Since TFLite does not support BatchMatMul operation, it unrolls a BatchMatMul // op into tf.Reshape, tf.Slice, tf.MatMul, tf.Pack, and tf.Reshape ops. void PopulateUnrollTfBatchMatMul(MLIRContext* context, RewritePatternSet& patterns); } // namespace TF } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 27 15:05:02 UTC 2022 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/basic.mlir
// CHECK-NEXT: [[ch1:%.*]], [[var:%.*]] = tfrt_fallback_async.executeop.seq([[in_chain]]) {{.*}} "tf.ReadVariableOp"([[arg1]]) // CHECK-NEXT: [[r0:%.*]] = tfrt_fallback_async.executeop {{.*}} "tf.MatMul"([[arg0]], [[var]]) %2 = "tf.MatMul"(%arg0, %1) {T = f32, _output_shapes = ["tfshape$dim { size: 3 } dim { size: 3 }"], device = "/device:CPU:0", transpose_a = false, transpose_b = false} : (tensor<3x1xf32>, tensor<1x3xf32>) -> tensor<3x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/keras_imagenet_main_graph_mode.golden_summary
BiasAdd 1 BiasAddGrad 1 Cast 3 Const 357 Conv2D 53 Conv2DBackpropFilter 53 Conv2DBackpropInput 52 DivNoNan 1 Equal 1 FusedBatchNorm 53 FusedBatchNormGrad 53 Identity 2 MatMul 3 MaxPool 1 MaxPoolGrad 1 Mean 1 Mul 164 Pad 1 ReadVariableOp 646 Relu 49 ReluGrad 49 Reshape 2 ResourceApplyKerasMomentum 161 ShapeN 50 Softmax 1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 06 10:38:14 UTC 2023 - 740 bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test_util.h
==============================================================================*/ #ifndef TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_ #define TENSORFLOW_C_EAGER_C_API_REMOTE_TEST_UTIL_H_ // Run a function containing a MatMul op and check its output. // If heavy_load_on_streaming_rpc is true, send some rpc requests before the one // which creates a remote input, to simulate a scenario that the remote input
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 11 22:56:03 UTC 2020 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/device_assignment.mlir
func.func @device_test(%arg0: tensor<3x1xf32>) -> (tensor<3x3xf32>) { // CHECK: device = "gpu" %0 = "tf.Const"() {value = dense<[[1.0, 2.0, 3.0]]> : tensor<1x3xf32>} : () -> tensor<1x3xf32> // CHECK: device = "gpu" %1 = "tf.MatMul"(%arg0, %0) {T = f32, _output_shapes = ["tfshape$dim { size: 3 } dim { size: 3 }"], device = "", transpose_a = false, transpose_b = false} : (tensor<3x1xf32>, tensor<1x3xf32>) -> tensor<3x3xf32> // CHECK: device = "cpu"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:47:26 UTC 2022 - 924 bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_quantized_functions_drq.mlir
// CHECK-NOT: func private @internal_quantize_i8 // CHECK-NOT: func private @internal_matmul_fn // CHECK: func private @quantized_matmul_fn // CHECK-SAME: tf_quant.quantized_ops = ["MatMul"] // CHECK: func private @quantized_conv2d_fn // CHECK-SAME: tf_quant.quantized_ops = ["Conv2D"] // CHECK: func private @quantized_depthwise_conv2d_fn // CHECK-SAME: tf_quant.quantized_ops = ["DepthwiseConv2D"]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Dec 01 12:06:54 UTC 2022 - 1K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/keras_imagenet_main.golden_summary
ArgMax 1 AssignAddVariableOp 1 BiasAdd 1 BiasAddGrad 1 Cast 115 Const 407 Conv2D 53 Conv2DBackpropFilter 53 Conv2DBackpropInput 52 Equal 1 FusedBatchNormGradV2 53 FusedBatchNormV2 53 MatMul 3 MaxPool 1 MaxPoolGrad 1 Mean 1 Mul 218 Pad 2 ReadVariableOp 538 Relu 49 ReluGrad 49 Reshape 2 ResourceApplyKerasMomentum 161 Slice 1 Softmax 1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 06 10:38:14 UTC 2023 - 874 bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir
// CHECK: %[[TRANSPOSE:.*]] = "tf.Transpose"(%[[DEQUANT]], %[[CST]]) : (tensor<3x4xf32>, tensor<?xi32>) -> tensor<*xf32> // CHECK: %[[MATMUL:.*]] = "tf.MatMul"(%arg0, %[[TRANSPOSE]]) <{grad_a = false, grad_b = false, transpose_a = false, transpose_b = true}> : (tensor<2x3xf32>, tensor<*xf32>) -> tensor<2x4xf32> // CHECK: return %[[MATMUL]] : tensor<2x4xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 59.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/analysis/tensor_array_side_effect_analysis.mlir
%x = "tf.TensorArrayReadV3"(%handle_0, %index, %flow_0) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<2x!tf_type.resource<tensor<?x100xf32>>>, tensor<i32>, tensor<f32>) -> tensor<?x100xf32> %y = "tf.MatMul"(%x, %cst) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<?x100xf32>, tensor<100x512xf32>) -> (tensor<?x512xf32>)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:41:10 UTC 2022 - 1.7K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/cpp_generator_test.cc
string output_dir = "tensorflow/c/experimental/ops/gen/cpp/golden"; string source_dir = "tensorflow"; string api_dirs = ""; std::vector<string> ops = { "Neg", // Simple unary Op "MatMul", // 2 inputs & attrs with default values "IdentityN", // Variadic input+output "SparseSoftmaxCrossEntropyWithLogits", // 2 outputs "AccumulatorApplyGradient", // 0 outputs
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 18 17:02:28 UTC 2022 - 2.9K bytes - Viewed (0)