- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 55 for bef_executor (0.35 sec)
-
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.cc
#include "tensorflow/core/tfrt/utils/fallback_tensor.h" #include "tfrt/bef/bef_buffer.h" // from @tf_runtime #include "tfrt/bef_converter/mlir_to_bef.h" // from @tf_runtime #include "tfrt/bef_executor/bef_file.h" // from @tf_runtime #include "tfrt/host_context/async_value.h" // from @tf_runtime #include "tfrt/host_context/chain.h" // from @tf_runtime #include "tfrt/host_context/execution_context.h" // from @tf_runtime
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/BUILD
[ "//third_party/tf_runtime_llvm:tfrt_translate_main", ], ["@tf_runtime//third_party/llvm_derived:tfrt_translate_main"], ), ) tf_cc_binary( name = "bef_executor", testonly = True, visibility = [":friends"], deps = [ "@tf_runtime//:dtype", "@tf_runtime//:simple_tracing_sink", "@tf_runtime//tools:bef_executor_expensive_kernels",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 19:04:21 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
return isa<tf_executor::IslandOp>(region->getParentOp()); } }; } // namespace TensorFlowExecutorDialect::TensorFlowExecutorDialect(MLIRContext *context) : Dialect(/*name=*/"tf_executor", context, TypeID::get<TensorFlowExecutorDialect>()) { addOperations< #define GET_OP_LIST #include "tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc.inc" >();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_validate_inputs.mlir
%control = tf_executor.island() wraps "tf.TPUReplicateMetadata"() {_tpu_replicate = "cluster", device = "/device:TPU:0", num_replicas = 2, topology = "topology"} : () -> () %ri, %c0 = tf_executor.island wraps "tf.TPUReplicatedInput"(%arg0, %arg1) {index = 1 : i64, is_mirrored_variable = false, is_packed = false} : (tensor<i32>, tensor<i32>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 06:51:01 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/convert_to_legacy_compile_and_replicate_attributes.mlir
// CHECK-NOT: _replication_info = %0 = tf_executor.graph { %outputs, %control = tf_executor.island wraps "tf.GuaranteeConst"(%arg1) {T = f32, device = ""} : (tensor<f32>) -> tensor<f32> %outputs_0, %control_1 = tf_executor.island wraps "tf.GuaranteeConst"(%arg2) {T = f32, device = ""} : (tensor<f32>) -> tensor<f32> %control_2 = tf_executor.island wraps "tf.NoOp"() {_pivot_for_cluster = "cluster", device = ""} : () -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
auto executor_graph = rewriter.create<tf_executor::GraphOp>(loc, func_type.getResults()); rewriter.inlineRegionBefore(graph.getNodes(), executor_graph.getBody(), executor_graph.getBody().end()); // Add terminator of tf_executor::graph rewriter.setInsertionPointToEnd(&executor_graph.getBody().front()); rewriter.create<tf_executor::FetchOp>(loc); // Add terminator of func
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
return llvm::isa<tf_executor::EnterOp, tf_executor::ExitOp, tf_executor::IslandOp, tf_executor::MergeOp, tf_executor::SwitchOp>(op); } // Assigns all data results to a specified device. void PopulateDeviceForOpResults( Operation& op, llvm::StringRef device, llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { Operation* op_to_update = &op;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/cluster_outlining.mlir
%4 = "tf.B"(%2) : (tensor<?xi32>) -> tensor<?xi32> tf_device.return %4 : tensor<?xi32> }) {} : () -> tensor<?xi32> // CHECK: tf_executor.yield %[[CLUSTER_OUTPUT]] tf_executor.yield %3 : tensor<?xi32> } tf_executor.fetch %1#0 : tensor<?xi32> } func.return %0 : tensor<?xi32> } // CHECK: func private @[[CLUSTER]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
let summary = "Lifts tf_executor.island inner ops from a tf_executor.graph"; let description = [{ This pass converts tf_executor.graphs consisting of only tf_executor.islands and a tf_executor.fetch into a sea of nodes consisting of TensorFlow Dialect ops by lifting such ops out of a tf_executor.graph's tf_executor.islands. If V1 control
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/tensorflow/tests/xla_validate_inputs.mlir
func.func @nested_entry_functions() attributes {tf.entry_function = {}} { tf_executor.graph { %control = tf_executor.island wraps "tf.StatefulPartitionedCall"() {config = "", config_proto = "", device = "/device:CPU:0", executor_type = "", f = @func} : () -> () tf_executor.fetch } func.return } func.func @func() attributes {tf.entry_function = {}} { func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 19:29:14 UTC 2024 - 818 bytes - Viewed (0)