- Sort Score
- Result 10 results
- Languages All
Results 11 - 12 of 12 for _disable_call_shape_inference (0.54 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
instead executes it on the device assigned to this op. }]; let arguments = (ins Variadic<TF_Tensor>:$args, FlatSymbolRefAttr:$f, DefaultValuedOptionalAttr<BoolAttr, "false">:$_disable_call_shape_inference ); let results = (outs Variadic<TF_Tensor>:$output ); let extraClassDeclaration = [{ // Gets the argument operands to the called function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK: %[[RESULT:.*]] = call @select_first(%arg0, %arg1) : (tensor<10x2xf32>, tensor<10x2xf32>) -> tensor<10x2xf32> %0 = "tf.LegacyCall"(%arg0, %arg1) {_disable_call_shape_inference = true, _tpu_replicate = "cluster", device = "", f = @select_first} : (tensor<10x2xf32>, tensor<10x2xf32>) -> tensor<10x2xf32> // CHECK: return %[[RESULT]] func.return %0: tensor<10x2xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)