- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for matmult (0.17 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py
] ) def matmul(self, matmul_input: core.Tensor) -> Mapping[str, core.Tensor]: """Performs a matrix multiplication. Args: matmul_input: Input tensor to matmul with the filter. Returns: A map of: output key -> output result. """ out = math_ops.matmul(matmul_input, self.matmul_filters) return {'output': out}
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/tests/legalize-tf.mlir
} func.func @matmul(%arg0: tensor<40x37xf32>, %arg1: tensor<37x40xf32>) -> tensor<40x40xf32> { %0 = "tf.MatMul"(%arg0, %arg1) {T = "tfdtype$DT_FLOAT", device = "/device:CPU:0", name = "MatMul", transpose_a = false, transpose_b = false} : (tensor<40x37xf32>, tensor<37x40xf32>) -> tensor<40x40xf32> func.return %0 : tensor<40x40xf32> // CHECK-LABEL: matmul
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
auto matmul = rewriter.create<TF::BatchMatMulV3Op>( loc, RankedTensorType::get(matmul_shape, result_type.getElementType()), lhs_flattend, rhs_flattend); if (result_type.hasStaticShape()) { auto reshaped = rewriter.create<mhlo::ReshapeOp>(loc, result_type, matmul.getResult()); return reshaped.getResult(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
``` The pass also works across control flow and functional calls. }]; } def UnrollBatchMatMulPass : Pass<"tf-unroll-batch-matmul", "mlir::func::FuncOp"> { let summary = "Unroll TF BatchMatMul op into Reshape, Slice, MatMul, Pack ops."; let constructor = "TF::CreateUnrollBatchMatMulPassPass()"; } def ClusterFormationPass : Pass<"tf-device-cluster-formation", "mlir::ModuleOp"> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
UniformQuantizedStableHloToTflPass> { private: void runOnOperation() override; }; // TODO: b/323645515 - Refactor reference functions. // Bias scales for matmul-like ops should be input scale * filter scale. Here it // is assumed that the input is per-tensor quantized and filter is per-channel // quantized. SmallVector<double> GetBiasScales(const double input_scale,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
if (fc_op.getFusedActivationFunction() != "NONE") return failure(); // Only fuse multiplier if all dimensions other than the depth dimension // are equal to 1 since otherwise // `matmul(x, filter) * cst != matmul(x, filter * cst)` // even if `filter` and `cst` are be broadcastable. auto shape = cst.getType().getShape(); if (!IsDimensionsDegenerateExceptLastOne(shape)) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir
func.return %0: tensor<2x3x7xf32> } // CHECK-LABEL: testBatchMatMulToMatMul func.func @testBatchMatMulToMatMul(%arg0: tensor<2x3xf32>, %arg1: tensor<3x2xf32>) -> tensor<2x2xf32> { // CHECK: %0 = "tf.MatMul"(%arg0, %arg1) <{grad_a = false, grad_b = false, transpose_a = false, transpose_b = false}> {device = "/job:localhost/replica:0/task:0/device:GPU:0"} : (tensor<2x3xf32>, tensor<3x2xf32>) -> tensor<2x2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 132.1K bytes - Viewed (0) -
CHANGELOG/CHANGELOG-1.5.md
* Support trust id as a scope in the OpenStack authentication logic ([#32111](https://github.com/kubernetes/kubernetes/pull/32111), [@MatMaul](https://github.com/MatMaul)) * Only wait for cache syncs once in NodeController ([#34851](https://github.com/kubernetes/kubernetes/pull/34851), [@ncdc](https://github.com/ncdc))
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Dec 24 02:28:26 UTC 2020 - 136.4K bytes - Viewed (0) -
src/encoding/json/testdata/code.json.gz
"min_t":1228859282,"max_t":1236205616,"mean_t":1232532449},{"name":"t0803-c5504-imrgn-l-02-b-ag-expected.png","kids":[],"cl_weight":0.0012228417427826135,"touches":2,"min_t":1228859282,"max_t":1236205616,"mean_t":1232532449},{"name":"t0905-c5525-fltmult-00-d-g-expected.checksum","kids":[],"cl_weight":0.0012228417427826135,"touches":2,"min_t":1228859282,"max_t":1236205616,"mean_t":1232532449},{"name":"t170602-bdr-conflct-w-39-d-expected.checksum","kids":[],"cl_weight":0.0012228417427826135,"touch...
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Fri Mar 25 04:02:36 UTC 2016 - 117.6K bytes - Viewed (0)