- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 148 for tpu0 (0.04 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/pywrap_quantize_model.cc
if (!exported_model.ok()) return exported_model.status(); // Remove the `tpu` tag from the debug quantized saved model as it is // for CPU. Note the 'tpu' value should be the same as `TPU` defined in // tensorflow/python/saved_model/tag_constants.py. if (quantization_options.has_debugger_config()) { tags.erase("tpu"); } py_function_library.SaveExportedModel(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 09 06:33:29 UTC 2024 - 12K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
mlir::ArrayAttr device_assignment_attr); // Finds the TPU compilation device and execution devices from `devices` for a // TPU computation subgraph. Compilation device is determined from looking up // all TPU_SYSTEM:0 devices and choosing the CPU device associated to the first // TPU_SYSTEM device sorted lexicographically by replica and task. Execution // devices are determined by looking up all TPU devices associated with each
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/legalize_tf_quant_test.cc
#include "tensorflow/core/framework/tensor_shape.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/protobuf/config.pb.h" #include "tensorflow/core/protobuf/tpu/compile_metadata.pb.h" #include "tensorflow/core/tpu/kernels/tpu_compile_op_support.h" #include "tsl/lib/core/status_test_util.h" namespace mlir::quant::stablehlo { namespace { using ::testing::Test;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 29 18:43:55 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_validate_inputs.mlir
// expected-warning @+1 {{TF2XLA TPU bridge input check: cluster op = tf.opA with cluster = cluster has successor as non cluster op tf.opB}} %out2, %c2 = tf_executor.island wraps "tf.opB"(%out) : (tensor<i32>) -> tensor<i32> // expected-error @+1 {{tf.TPUReplicatedOutput' op TF2XLA TPU bridge input check: non-cluster op = tf.opB has invalid successor op = tf.TPUReplicatedOutput}}
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/tf2xla/internal/utils/test_metadata_config.cc
#include "tensorflow/core/protobuf/tpu/compile_metadata.pb.h" #include "tsl/platform/errors.h" namespace tensorflow { namespace tf2xla { namespace internal { namespace { constexpr char kEntryFuncName[] = "main"; absl::Status SetupArguments(mlir::ModuleOp module, std::vector<TensorShape>& arg_shapes, tpu::TPUCompileMetadataProto& metadata_proto) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
arg->set_dtype(dtype); // TODO(lyandy): Support other arg kinds. if (dtype == tensorflow::DT_RESOURCE) arg->set_kind(tensorflow::tpu::TPUCompileMetadataProto::Arg::VARIABLE); else arg->set_kind(tensorflow::tpu::TPUCompileMetadataProto::Arg::PARAMETER); // Populate argument shapes. *arg->mutable_shape() = tensorflow::TensorShapeProto(); if (auto ranked_tensor_type =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
// TPU-annotated operations and intended to preserve backward compatibility with // TFv1. struct TpuV1BridgeExecutorIslandCoarsening : public impl::TpuV1BridgeExecutorIslandCoarseningPassBase< TpuV1BridgeExecutorIslandCoarsening> { void runOnOperation() override; }; // Returns name of TPU cluster, if op belongs to a TPU cluster. Otherwise, // returns `std::nullopt`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_to_hlo_test.cc
#include "tensorflow/core/protobuf/tpu/compile_metadata.pb.h" #include "tensorflow/core/tpu/kernels/tpu_compile_op_support.h" #include "tsl/platform/statusor.h" namespace tensorflow { namespace tf2xla { namespace internal { using ::tensorflow::monitoring::testing::CellReader; using tpu::MlirToHloArgs; using tpu::ShardingAndIndex; using tpu::TPUCompileMetadataProto;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph_test.cc
namespace tensorflow { namespace tf2xla { namespace v1 { namespace { using ::tensorflow::monitoring::testing::CellReader; using ::tensorflow::tpu::FunctionToHloArgs; using ::tensorflow::tpu::MlirToHloArgs; using ::tensorflow::tpu::ShardingAndIndex; using ::tsl::monitoring::testing::Histogram; static constexpr char kCompilationTimeStreamzName[] = "/tensorflow/core/tf2xla/api/v1/phase2_compilation_time";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:08:57 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info_test.cc
#include "tensorflow/core/protobuf/error_codes.pb.h" #include "tensorflow/core/tfrt/common/create_pjrt_client_util.h" #include "tensorflow/core/tfrt/common/pjrt_util.h" #include "tensorflow/core/tpu/tpu_defs.h" namespace tensorflow { namespace { using XlaDeviceCompiler = DeviceCompiler<xla::LocalExecutable, xla::LocalClient>; using PjRtDeviceCompiler =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Jan 14 15:17:12 UTC 2024 - 13.6K bytes - Viewed (0)