- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for tpu_subgraph (0.1 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_tail_with_tobool_op.mlir
func.return %arg0 : tensor<*xi1> } func.func private @if_else_branch(%arg0: tensor<*xi1>, %arg1: tensor<*xi64>, %arg2: tensor<*xi64>) -> tensor<*xi1> { func.return %arg0 : tensor<*xi1> } func.func @"tpu_subgraph"(%arg0: tensor<*xi64>) -> tensor<*xi1> { %cst = "tf.Const"() {value = dense<0> : tensor<i64>} : () -> tensor<i64> %cst_0 = "tf.Const"() {value = dense<0> : tensor<i32>} : () -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
@function.Defun(capture_resource_var_by_value=False) def tpu_subgraph(): return tf.tpu.rewrite(model_fn, args) return tf.raw_ops.TPUPartitionedCall( args=tpu_subgraph.captured_inputs, Tout=[o.type for o in tpu_subgraph.definition.signature.output_arg], f=tpu_subgraph, device_ordinal=[0])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0)