- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 20 for tpu0 (0.04 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
} def TPUPartitionedOpConversionPass : Pass<"tf-tpu-partitioned-op-conversion", "mlir::func::FuncOp"> { let summary = "Rewrite all TPU Partitioned ops into their V2 counterparts."; let constructor = "TFTPU::CreateTPUPartitionedOpConversionPass()"; } def TPUValidateInputsPass : Pass<"tf-tpu-validate-inputs", "ModuleOp"> { let summary = "Validates inputs to the TPU TF/XLA bridge"; let description = [{
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/tf2xla/api/v1/compile_tf_graph.cc
namespace tensorflow { namespace tf2xla { namespace v1 { using ::tensorflow::tpu::FunctionToHloArgs; using ::tensorflow::tpu::GuaranteedConsts; using ::tensorflow::tpu::MlirToHloArgs; using ::tensorflow::tpu::ShardingAndIndex; auto* phase2_bridge_compilation_status = tensorflow::monitoring::Counter<1>::New( "/tensorflow/core/tf2xla/api/v1/" "phase2_compilation_status", /*metric_name*/
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 14K 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/tf2xla/api/v1/BUILD
"//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core/tpu:tpu_compile", "//tensorflow/core/tpu/kernels:tpu_compile_op_support", "//tensorflow/core/tpu/kernels:tpu_compile_proto_cc", "//tensorflow/core/tpu/kernels:tpu_util", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/log", "@com_google_absl//absl/status",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 12.7K bytes - Viewed (0) -
.bazelrc
# ARM64 WHEEL test:linux_arm64_wheel_test_filters --test_tag_filters=-no_oss,-no_aarch64,-oss_excluded,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_oss_py38,-no_oss_py39,-no_oss_py310
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 17:12:54 UTC 2024 - 52.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/BUILD
"//tensorflow/core:lib", "//tensorflow/core/platform:statusor", "//tensorflow/core/tpu:tpu_compile", "//tensorflow/core/tpu/kernels:tpu_compile_op_support", "//tensorflow/core/tpu/kernels:tpu_compile_proto_cc", "//tensorflow/core/tpu/kernels:tpu_util_hdrs", "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/strings",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
CreateTFExecutorIslandCoarseningPass(); // Creates a pass to merge IslandOps for operation marked for execution on TPU. // This is a V1 backward compatibility. std::unique_ptr<OperationPass<ModuleOp>> CreateTFExecutorTPUV1IslandCoarseningPass(); // Creates a pass to outlining TPU clusters from single IslandOp into a nested // module suitable for being processed as-if it was a V2 module.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf_test.cc
namespace tensorflow { namespace tf2xla { namespace v2 { using ::tensorflow::monitoring::testing::CellReader; using ::testing::Not; using ::testing::TestWithParam; using tpu::FunctionToHloArgs; using tpu::MlirToHloArgs; using tpu::ShardingAndIndex; using tpu::TPUCompileMetadataProto; static constexpr char kCompilationTimeStreamzName[] = "/tensorflow/core/tf2xla/api/v2/phase2_compilation_time";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 16.1K bytes - Viewed (0) -
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)