- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 215 for tpu0 (0.72 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc
#include "tensorflow/core/platform/profile_utils/cpu_utils.h" #include "tensorflow/core/platform/types.h" #include "tensorflow/core/protobuf/tpu/compile_metadata.pb.h" #include "tensorflow/core/tpu/kernels/tpu_compile_op_support.h" #include "tensorflow/core/tpu/tpu_compile.h" #include "tsl/platform/error_logging.h" #include "tsl/platform/errors.h" #include "tsl/platform/statusor.h" namespace tensorflow {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/function/function.h
// If true, use ServingCoreSelector to pick TPU core. Otherwise, obtain core // location from assigned device name. // Currently we don't use core_selector for training use cases. bool tpu_use_core_selector = false; // If true, use BundledTransferToTpuOp to transfer variables and input tensors // to TPU. bool tpu_use_bundled_transfer = false; // If true, lower an TF op that's placed on TPU device to be executed with
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/transforms/set_tpu_infeed_layout.cc
#include "xla/shape.h" #include "xla/stream_executor/tpu/c_api_conversions.h" #include "xla/stream_executor/tpu/tpu_api.h" #include "xla/translate/mhlo_to_hlo/type_to_shape.h" namespace mlir { static FailureOr<std::vector<int64_t>> GetTPUInfeedLayoutFromAPI( RankedTensorType t) { // Call the TPU API to determine the right infeed layout. Note that // this can fail if we're not running on a TPU-enabled node.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.h
// client - The Xla Compilation client. absl::StatusOr<tensorflow::XlaCompilationResult> LegalizeMlirToHlo( const std::variant<tpu::MlirToHloArgs, tpu::FunctionToHloArgs>& computation, const tpu::TPUCompileMetadataProto& metadata, bool use_tuple_args, llvm::StringRef device_type, std::vector<std::unique_ptr<mlir::Pass>>& custom_legalization_passes,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 07:32:57 UTC 2024 - 2.9K bytes - Viewed (0) -
istioctl/pkg/metrics/metrics.go
return time.Duration(val) * time.Millisecond } func printHeader(writer io.Writer) { w := tabwriter.NewWriter(writer, 13, 1, 2, ' ', tabwriter.AlignRight) _, _ = fmt.Fprintf(w, "%40s\tTOTAL RPS\tERROR RPS\tP50 LATENCY\tP90 LATENCY\tP99 LATENCY\t\n", "WORKLOAD") _ = w.Flush() } func printMetrics(writer io.Writer, wm workloadMetrics) { w := tabwriter.NewWriter(writer, 13, 1, 2, ' ', tabwriter.AlignRight)
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Sat Apr 13 05:23:38 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/legalize_tf.cc
return absl::OkStatus(); } } // namespace absl::StatusOr<tensorflow::XlaCompilationResult> LegalizeMlirToHlo( const std::variant<tpu::MlirToHloArgs, tpu::FunctionToHloArgs>& computation, const tpu::TPUCompileMetadataProto& metadata, bool use_tuple_args, llvm::StringRef device_type, std::vector<std::unique_ptr<mlir::Pass>>& custom_legalization_passes,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 00:40:46 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/replicate_to_island.mlir
tf_executor.fetch } func.return } // CHECK: "tf.opA" // device = "/TPU:0" // CHECK: _parallel_execution_ids = "r0:0" // CHECK: "tf.opA" // device = "/TPU:0" // CHECK: _parallel_execution_ids = "r0:1" // CHECK: "tf.opA" // device = "/TPU:1" // CHECK: _parallel_execution_ids = "r1:0" // CHECK: "tf.opA" // device = "/TPU:1"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 15.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir_test.cc
#include "tensorflow/core/lib/monitoring/cell_reader.h" #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 { namespace { using tpu::MlirToHloArgs; using tpu::ShardingAndIndex; using tpu::TPUCompileMetadataProto; static constexpr char kMlirModuleStr[] = R"(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/core/tpu/tpu_defs.h" namespace mlir { namespace quant { namespace { #include "tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.inc" // Convert a TPU model to be compatible on CPU by rewriting/removing TPU ops. class ConvertTpuModelToCpuPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
ci/official/containers/linux_arm64/devel.usertools/aarch64.bazelrc
# bazel test invocation as normal. test:nonpip_filters --test_tag_filters=-no_oss,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_aarch64,-no_oss_py38,-no_oss_py39,-no_oss_py310 test:nonpip_filters --build_tag_filters=-no_oss,-oss_serial,-gpu,-tpu,-benchmark-test,-v1only,-no_aarch64,-no_oss_py38,-no_oss_py39,-no_oss_py310
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 12:25:39 UTC 2023 - 5.8K bytes - Viewed (0)