- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 245 for tpu0 (0.04 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/utils/BUILD
"//tensorflow/compiler/tf2xla/kernels:xla_ops", "//tensorflow/core:framework", "//tensorflow/core:test_main", "//tensorflow/core/protobuf/tpu:compile_metadata_proto_cc", "//tensorflow/core/tpu/kernels/xla:host_compute_ops", "@com_google_absl//absl/status", "@com_google_absl//absl/strings:string_view", "@llvm-project//mlir:FuncDialect", "@llvm-project//mlir:IR",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/rewrite_ifrt_load_variable.mlir
// RUN: tf-tfrt-opt -split-input-file -tf-mlrt-rewrite-ifrt-load-variable %s | FileCheck %s // Variable is used by both CPU and TPU // // CHECK-LABEL: func @serving_default(%arg0: tensor<1x3xf32>) -> tensor<1x1xf32> // CHECK-NEXT: [[HANDLE:%.*]] = "tf.VarHandleOp"() // CHECK-NEXT: [[ARRAYKEY:%.*]], [[FURTURE:%.*]] = "tf_mlrt.tf_ifrt_load_variable"([[HANDLE]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:35:32 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_sequencing.mlir
return %0 : tensor<i1> } // Generated functions // non_tpu should have to TPU ops - just identity and return (in this test). // CHECK: func.func private @_func_non_tpu // CHECK-NEXT: tf.Identity // CHECK-NEXT: return // sc_forward should have TPU ops including replicated outputs but not inputs // CHECK: func.func private @_func_sc_forward // CHECK-NOT: TPUReplicatedInput
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 01 21:27:49 UTC 2023 - 19.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_pipelining.mlir
%cst_12 = "tf.Const"() {_replication_info = "repl_info", _xla_compile_device_type = "TPU", device = "", value = dense<1> : tensor<i64>} : () -> tensor<i64> "tf.AssignAddVariableOp"(%37, %cst_12) {_has_manual_control_dependencies = true, _replication_info = "while/cluster_while_body_451", _xla_compile_device_type = "TPU", device = ""} : (tensor<*x!tf_type.resource<tensor<i64>>>, tensor<i64>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 33.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tpu_embedding_ops_registry.h
#include "llvm/ADT/DenseSet.h" #include "mlir/IR/Operation.h" // from @llvm-project #include "mlir/Support/TypeID.h" // from @llvm-project namespace mlir { namespace TF { // A global ops registry that is used to hold TPU embedding ops. // // Example: // TPUEmbeddingOpsRegistry::Global().Add<TF::FooOp>(); // for (auto op_type_id : TPUEmbeddingOpsRegistry::Global().GetOpsTypeIds()) // { // ... // }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 14 05:42:28 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
// For all other inputs that go from TPU op to TPU op, insert the // TPUOutput/Input pair. // Add TPUReplicatedInput/TPUReplicatedOutput pairs along each edge. // TODO(bfontain): Should be merged with the above loop. llvm::SetVector<Value> values_to_add_nodes; for (Operation* op : backward_pass_ops) { // TODO(bfontain): Should validate that all the TPU ops are in the same // replication region.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_colocate_splits.mlir
// RUN: tf-opt %s -split-input-file -verify-diagnostics -tf-tpu-colocate-splits | FileCheck %s // CHECK-LABEL: func @colocate_split_with_pred func.func @colocate_split_with_pred() { // CHECK: Split // CHECK-SAME: _class = ["loc:@class"] tf_executor.graph { %c, %control0 = tf_executor.island wraps "tf.Const"() {value = dense<0> : tensor<i32>} : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 18:44:34 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_inlining/executor_tpuv1_inline_tpu_island.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 11:03:04 UTC 2022 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compiler_options_util.h
const FunctionLibraryRuntime& function_library, DeviceBase* device, se::Stream* stream, const XlaPlatformInfo& platform_info, bool has_ref_vars); // Returns created options for XLA compiler when TFRT-TPU is used. XlaCompiler::Options GenerateCompilerOptionsForTfrtTpu( const DeviceCompiler<xla::LocalExecutable, xla::LocalClient>& xla_device_compiler, const FunctionLibraryRuntime& function_library);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 29 01:41:20 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h
// // Duplicating shape-determining constants is required to place constants that // affect the shape of a tensor to be placed in the TPU graph instead of in the // CPU graph, when the graph gets converted for TPU inference. This allows these // constants to be known at XLA compilation time. void AddExportPasses(mlir::PassManager& pm, bool duplicate_shape_determining_constants);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 6.9K bytes - Viewed (0)