- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 116 for PartitionedCall (0.17 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
Option<bool> skip_partitioned_calls_{ *this, "skip-partitioned-calls", ::llvm::cl::desc( "Skip tf.StatefulPartitionedCall and tf.PartitionedCall")}; }; void TFToMhloPass::runOnOperation() { auto func = getOperation(); MLIRContext *context = func->getContext(); RewritePatternSet patterns(context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_xla.mlir
// CHECK-NEXT: [[q_bias:%.+]] = "quantfork.qcast"([[bias]]) : (tensor<2xf32>) -> tensor<2x!quant.uniform<i32:f32, 0.044022349891595126>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util_test.cc
*library.add_function() = identity_func; Output in = ops::Placeholder(root, DT_FLOAT); NameAttrList b_name_attr; b_name_attr.set_name("IdentityFunc"); ops::PartitionedCall call(root.WithOpName("call"), {in}, {DT_FLOAT}, b_name_attr); GraphDef graph_def; TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); TF_ASSERT_OK(root.ToGraphDef(&graph_def));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 22.3K bytes - Viewed (0) -
tensorflow/cc/gradients/functional_grad.cc
auto grad = SymbolicGradient(scope, func_inputs, input_dtypes, f); for (int i = 0; i < num_inputs; i++) { grad_outputs->push_back(grad[i]); } return scope.status(); } REGISTER_GRADIENT_OP("PartitionedCall", PartitionedCallGrad); REGISTER_GRADIENT_OP("StatefulPartitionedCall", PartitionedCallGrad); } // anonymous namespace } // namespace ops
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 20:09:06 UTC 2021 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/remove_unused_arguments.mlir
return %arg0 : tensor<f32> } // CHECK-LABEL: handles_partitioned_function_calls func.func @handles_partitioned_function_calls(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> { // CHECK: PartitionedCall"() %1 = "tf.PartitionedCall"(%arg0, %arg1) {f = @f} : (tensor<f32>, tensor<f32>) -> tensor<f32> return %1 : tensor<f32> } // ----- func.func private @f(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 06 23:00:44 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/graph-as-function.mlir
%2:2 = tf_executor.island wraps "tf.StatefulPartitionedCall"(%0#0, %arg1) {Tin = ["tfdtype$DT_FLOAT", "tfdtype$DT_RESOURCE"], Tout = ["tfdtype$DT_FLOAT"], _gradient_op_type = "PartitionedCall-1205", config = "", config_proto = "\0A\07\0A\03GPU\10\00\0A\07\0A\03CPU\10\012\02J\008\01", device = "", executor_type = "", f = @function0} : (tensor<f32>, tensor<*x!tf_type.resource<tensor<3x3x1x32xf32>>>) -> tensor<f32> loc("statefulpartitionedcall")...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 25 12:28:56 UTC 2022 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
: (tensor<!tf_type.variant<tensor<f32>>>, tensor<i1>) -> tensor<!tf_type.variant<tensor<f32>>> // CHECK: %[[CALL2:.*]]:2 = "tf.PartitionedCall"(%[[INIT]], // CHECK-SAME: f = @callee_tensorlist_decomposed %call2 = "tf.PartitionedCall"(%tl, %arg0) {f = @callee, config = "", config_proto = "", executor_type = ""} : (tensor<!tf_type.variant<tensor<f32>>>, tensor<i1>) -> tensor<!tf_type.variant<tensor<f32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util_test.cc
} void CreateSubgraphCallingFunctionWithRefVar(const Scope& s) { NameAttrList ref_float_function; ref_float_function.set_name("RefFloatFn"); ops::PartitionedCall call(s.WithOpName("RefFloat"), {absl::Span<Input>{}}, {DT_FLOAT}, ref_float_function); Output constant = ops::Const(s.WithOpName("constant_ref_pco"), Input::Initializer(0.0));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/cc/gradients/functional_grad_test.cc
Output x = Placeholder(scope_, DT_FLOAT); NameAttrList f; f.set_name("XTimesTwo"); (*f.mutable_attr())["T"].set_type(DT_FLOAT); auto results = PartitionedCall(scope_, std::initializer_list<Input>{x}, {DT_FLOAT}, f); RunTest(x, {}, results[0], {}); auto stateful_results = StatefulPartitionedCall( scope_, std::initializer_list<Input>{x}, {DT_FLOAT}, f);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 26 14:42:24 UTC 2021 - 3K bytes - Viewed (0)