- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 6,431 for runAction (0.23 sec)
-
tensorflow/compiler/mlir/tfrt/function/function.cc
See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ #include "tensorflow/compiler/mlir/tfrt/function/function.h" #include "absl/log/log.h" #include "absl/strings/match.h" #include "mlir/IR/OperationSupport.h" // from @llvm-project #include "mlir/Pass/PassManager.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 08:13:15 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/function/function.h
// to proactively transfer the result to host since the consumer op (or // function) of the result may still be on TPU. // TODO(b/194081364): remove this option once we unify servo TPU serving // result transfer behavior. bool tpu_transfer_result_to_host = false; }; // Compile MLIR generated by tf.function in TF dialect into BEF. Status CompileTFMLIRToBEF(const TfrtFunctionCompileOptions& options,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 04:50:20 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/function-order.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 17 18:52:47 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/function-func-attr.pbtxt
} node { name: "custom_embedding_matmul_func_call" op: "custom_embedding_matmul" } library { function { signature { name: "custom_relu" } attr { key: "_implements" value { func { name: "tensorflow.relu" } } } } function { signature { name: "custom_embedding_matmul" } attr { key: "_implements"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 01 20:09:54 UTC 2023 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-as-function.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-graph-as-function -tf-xla-compile-device-type="GPU" -tf-enable-soft-placement-on-import=true -o - | FileCheck %s # Verify main graph was converted to a function, args/rets are mapped correctly, # and ops in the main graph are retained. In addition, check if subsequent # functions are converted.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 24 00:18:34 UTC 2023 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/graph-as-function.mlir
// RUN: tf-mlir-translate -mlir-to-graphdef %s -tf-graph-as-function -o - | FileCheck %s func.func @main(%arg0: tensor<*x!tf_type.resource>, %arg1: tensor<*x!tf_type.resource<tensor<3x3x1x32xf32>>>, %arg2: tensor<*xf32>, %arg3: tensor<2x4x6x8xi32>) -> (tensor<f32>, tensor<f32>) attributes {tf.entry_function = {inputs = "args_0,args_1,args_2,args_3", outputs = "rets_0_RetVal,rets_1_RetVal"}} { %graph:2 = tf_executor.graph {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-defs.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -o - | FileCheck %s # Verify that we properly import call site function attributes. # CHECK: tf.If # CHECK-SAME: then_branch = @ # CHECK-SAME: then_branch.how_many = 32 # CHECK-SAME: then_branch.ping = "ack" node { name: "Placeholder" op: "Placeholder" attr { key: "dtype" value { type: DT_INT32 } } attr {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 24 00:20:25 UTC 2020 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-call.pbtxt
attr { key: "_disable_call_shape_inference" value { b: true } } attr { key: "_tpu_replicate" value { s: "cluster" } } } library { function { signature { name: "test_func_name" input_arg { name: "a_0" type: DT_INT32 } output_arg { name: "a" type: DT_INT32 } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-function-name-bug.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/function-resource-args.mlir
// resource_arg_unique_id. // CHECK: name: "x" // CHECK: op: "VarHandleOp" // CHECK: name: "func_call" // CHECK: input: "called" // CHECK: library // CHECK: function // CHECK: signature // CHECK: input_arg // CHECK: type: DT_RESOURCE // CHECK: input_arg // CHECK: type: DT_RESOURCE // CHECK: output_arg
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 2.6K bytes - Viewed (0)