- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for func_1_cpu (0.13 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/common/subgraph.h
namespace tac { // Interface name here is the "hook" between the CallOp and FuncOps. // Take the following example: // // call @func_1_CPU {tac.interface_name = "func_1"} // // "func_1" is the interface name where "func_1_cpu" is the real implementation // we can have multiple FuncOps like "func_1_cpu" and "func_1_gpu" and they // both implement "func_1". // // The attribute on the FuncOp means what it actually implements while the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 28 05:18:47 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/compute-cost.mlir
func.func @func_0_GPU(%arg0: tensor<10x10x10xf32>, %arg1: tensor<f32>) -> tensor<10x10x10xf32> attributes {tac.device = "GPU", tac.interface_name = "func_0"} { %0 = "tfl.add"(%arg0, %arg1) {fused_activation_function = "RELU", tac.device = "GPU"} : (tensor<10x10x10xf32>, tensor<f32>) -> tensor<10x10x10xf32> func.return %0 : tensor<10x10x10xf32> } // ----- // CHECK: tac.cost = 2.000000e+03
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:29:10 UTC 2022 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tests/get-op-cost.mlir
func.func @func_0_CPU(%arg0: tensor<10x10x10xf32>, %arg1: tensor<10xf32>) -> tensor<10x10x10xf32> attributes {tac.device = "CPU", tac.interface_name = "func_0"} { // CHECK: tac.cost = 1.000000e+03 %0 = "tfl.add"(%arg0, %arg1) {fused_activation_function = "RELU", tac.device = "CPU"} : (tensor<10x10x10xf32>, tensor<10xf32>) -> tensor<10x10x10xf32> func.return %0 : tensor<10x10x10xf32> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 24 05:29:10 UTC 2022 - 5.7K bytes - Viewed (0)