- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for tpu0 (0.27 sec)
-
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/tensorflow/tests/tpu_rewrite.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir
module attributes {tf.versions = {producer = 888 : i32}, tf.devices = ["/job:worker/replica:0/task:0/device:CPU:0", "/job:worker/replica:0/task:0/device:TPU_SYSTEM:0", "/job:worker/replica:0/task:0/device:TPU:0"]} { // Tests that TPU cluster with no outside compilation does not generate parallel_execute. // CHECK-LABEL: func @no_outside_compilation func.func @no_outside_compilation() -> tensor<2xi32> { // CHECK-NOT: "tf_device.parallel_execute"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 129.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// CHECK: return %0 : tensor<?x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
pkg/scheduler/internal/queue/scheduling_queue_test.go
q := NewTestQueue(ctx, newDefaultQueueSort(), WithClock(c)) // Add a few pods to priority queue. for i := 0; i < 5; i++ { p := st.MakePod().Name(fmt.Sprintf("test-pod-%v", i)).Namespace("ns1").UID(fmt.Sprintf("tp00%v", i)).Priority(highPriority).Node("node1").NominatedNodeName("node1").Obj() q.Add(logger, p) } c.Step(time.Microsecond) // Simulate a pod being popped by the scheduler, determined unschedulable, and
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Jun 12 13:26:09 UTC 2024 - 146.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
} func.return // expected-remark@above {{ID: 6}} // expected-remark@above {{Sinks: {5}}} } // ----- // Tests that we don't create dependencies between TPU compile ops. func.func @tpu_compile_ops( // expected-remark@above {{ID: 7}} %arg0: tensor<!tf_type.string>, %arg1: tensor<!tf_type.string>) { tf_executor.graph { // expected-remark@above {{ID: 5}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0)