- Sort Score
- Result 10 results
- Languages All
Results 171 - 180 of 245 for tpu0 (0.73 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" // Pass that co-locates resource ops that use composite device resources // (packed tensors) with the underlying physical TPU device. struct TPUColocateCompositeResourceOps : public impl::TPUColocateCompositeResourceOpsPassBase< TPUColocateCompositeResourceOps> { void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
auto status_or_device_coodinates = tensorflow::GetDeviceCoordinates(device_assignment_attr); if (!status_or_device_coodinates.ok()) return cluster.emitError() << "error in fetching tpu device coordinates: " << status_or_device_coodinates.status().message(); status_or_tpu_device_assignment = tensorflow::GetTPUCompilationAndExecutionDevices(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util_test.cc
#include "tensorflow/compiler/jit/flags.h" #include "tensorflow/compiler/tf2xla/xla_compiler.h" #include "tensorflow/core/framework/fake_input.h" #include "tensorflow/core/kernels/ops_testutil.h" #include "tensorflow/core/tpu/tpu_defs.h" namespace tensorflow { namespace { TEST_F(OpsTestBase, CreateSingleOpGraph) { TF_EXPECT_OK(NodeDefBuilder("identity_op", "Identity") .Input(FakeInput(DT_FLOAT))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 21:48:05 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
RewritePatternSet* patterns) { for (auto op : context->getRegisteredOperations()) op.getCanonicalizationPatterns(*patterns, context); } // Adds the list of ops that are supported on TPU through constant folding which // may depend on the inputs shapes not known at this point. Such ops may not // have any legalization or canonicalization patterns but shouldn't be marked // for outside compilation. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/transforms.cc
// TODO: b/230572023 - Consider improving shape inference for While op instead // of dropping the attribute. This need not be correct for models not trained // on TPU. // Optimizes TF graph via cleanups, merges, rewrites, constant folding, // and edge case handling where possible. pm.addNestedPass<func::FuncOp>(TF::CreateDropWhileShapeInvariantPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 04:34:23 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_update_embedding_enqueue_op_inputs.mlir
// RUN: tf-opt %s -split-input-file -verify-diagnostics -tf-tpu-update-embedding-enqueue-op-inputs | FileCheck %s // CHECK-LABEL: func @check_enqueue_ops_update_for_eval // CHECK-SAME: %[[ARG_0:[a-z0-9]*]]: tensor<?x2xi32> // CHECK-SAME: %[[ARG_1:[a-z0-9]*]]: tensor<?x2xi32> // CHECK-SAME: %[[ARG_2:[a-z0-9]*]]: tensor<?x2xi32> // CHECK-SAME: %[[ARG_3:[a-z0-9]*]]: tensor<?xi32> // CHECK-SAME: %[[ARG_4:[a-z0-9]*]]: tensor<?xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/extract_tpu_copy_with_dynamic_shape_op.mlir
// RUN: tf-opt -split-input-file -verify-diagnostics -tf-extract-tpu-copy-with-dynamic-shape-op %s | FileCheck %s // Test that extract TPUCopyWithDynamicShape from host launch to device launch. module attributes {tf.devices = {"/job:localhost/replica:0/task:0/device:COMPOSITE:0", "/job:localhost/replica:0/task:0/device:CPU:0", "/job:localhost/replica:0/task:0/device:TPU:0", "/job:localhost/replica:0/task:0/device:TPU_SYSTEM:0"}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 4.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/quantize_model.cc
quantization_config.mutable_static_range_ptq_preset() ->set_enable_per_channel_quantized_weight( quantization_options.enable_per_channel_quantization()); // When targeting server TPUs quantized types should be unpacked into // integer ops. quantization_config.mutable_pipeline_config()->set_unpack_quantized_types( true); *quantization_config.mutable_debugger_config() =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
// Get the device context given the index. absl::StatusOr<DeviceContext*> GetDeviceContextWithIndex(int index); // Instructs this XlaDevice to set a AcceleratorDeviceInfo, which holds extra // information for GPU and TPU devices. Status UseAcceleratorDeviceInfo() TF_LOCKS_EXCLUDED(mu_); // Instructs this XlaDevice to return 'sync_on_completion' for // AllowsSyncOnCompletion().
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/bridge_logger.h
#include "mlir/Pass/PassManager.h" // from @llvm-project #include "mlir/Support/Timing.h" // from @llvm-project namespace tensorflow { // Logger for logging MLIR modules before and after passes in MLIR TPU bridge. // // The IR logging can be restricted to a particular set of pass invocations via // filters that are specified with the `MLIR_BRIDGE_LOG_PASS_FILTER` and // `MLIR_BRIDGE_LOG_STRING_FILTER` environment variables.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 06 22:29:51 UTC 2023 - 4.9K bytes - Viewed (0)