- Sort Score
- Result 10 results
- Languages All
Results 11 - 14 of 14 for StatefulPartitionedCall (0.33 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
result.replaceAllUsesWith(if_op.getOperand(then_aliased_input + 1)); } } if_op.replaceAllUsesWith(new_if); if_op.erase(); return success(); } // Handles stack usage by a tf.StatefulPartitionedCall or a tf.PartitionedCall. // It will first check if the callee was previously handled, and try to reuse // that result if so. Otherwise, it will clone and convert the callee function,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
// CHECK: [[results_0:%.*]] = call @callee( // CHECK-SAME: (!tf_mlrt.tensor) -> !tf_mlrt.tensor %1 = "tf.StatefulPartitionedCall"(%0) {config = "", config_proto = "", executor_type = "", f = @callee} : (tensor<i32>) -> (tensor<i32>) // CHECK-NEXT: [[results_1:%.*]] = call @callee( // CHECK-SAME: (!tf_mlrt.tensor) -> !tf_mlrt.tensor
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
arg_buffer_size_is_fixed(i)}; } } UpdateFuncType(func); } // Holds information about a decomposed callee function for // PartitionedCall/StatefulPartitionedCall. struct PartitionedCallDecompositionInfo { bool signature_change; func::FuncOp decomposed_callee; llvm::SmallDenseMap<int64_t, int64_t> buffer_arg_to_size_arg;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
// (checked later). if (auto device_type_attr = op.getAttrOfType<StringAttr>(mlir::TF::kCompileDeviceTypeAttr)) { // tf.StatefulPartitionedCall ops with and without // _tpu_replicate attributes may exist in the same graph. Ops without // the attribute but with _XlaMustCompile=true would have // _xla_compile_device_type="" after
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0)