- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for WithOpName (0.2 sec)
-
tensorflow/compiler/jit/shape_inference_test.cc
auto a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT, ops::Placeholder::Shape({2, 3})); auto b = ops::Placeholder(root.WithOpName("B"), DT_FLOAT, ops::Placeholder::Shape({3})); auto c = ops::Placeholder(root.WithOpName("C"), DT_FLOAT); auto d = ops::Add(root.WithOpName("D"), a, b); auto e = ops::Add(root.WithOpName("E"), d, c);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass_test.cc
auto a = ops::Placeholder(scope.WithOpName("A"), DT_INT32); auto b = ops::Placeholder(scope.WithOpName("B"), DT_FLOAT); auto c = ops::Placeholder(scope.WithOpName("C"), DT_INT32); auto d = ops::Placeholder(scope.WithOpName("D"), DT_FLOAT); auto u = ops::Placeholder(scope.WithOpName("U"), DT_RESOURCE); auto v = ops::Placeholder(scope.WithOpName("V"), DT_RESOURCE);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 18:03:15 UTC 2023 - 14.7K bytes - Viewed (0) -
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_RESOURCE, 0); Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_RESOURCE, 1); Output arg2 = ops::_Arg(s.WithOpName("arg2"), DT_INT32, 2); auto ret0 = ops::_Retval(s.WithOpName("ret0"), arg1, 0); auto ret1 = ops::_Retval(s.WithOpName("ret1"), arg0, 1); auto ret2 = ops::_Retval(s.WithOpName("ret2"), arg2, 2);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cluster_util_test.cc
Output a = ops::Const(root.WithOpName("a"), Input::Initializer(0.0)); Output enter_0 = ops::internal::Enter(root.WithOpName("enter_0"), a, "frame_0"); Output exit_0 = ops::internal::Exit(root.WithOpName("exit_0"), enter_0); Output enter_1 = ops::internal::Enter(root.WithOpName("enter_1"), a, "frame_1"); Output exit_1 = ops::internal::Exit(root.WithOpName("exit_1"), enter_1);
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/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc
Output input = ops::Placeholder(root.WithOpName("input"), DT_FLOAT); Output begin = ops::Placeholder(root.WithOpName("begin"), DT_INT32); Output size = ops::Const(root.WithOpName("size"), {-1, 500}); Output control_pred = ops::Placeholder(root.WithOpName("control"), DT_BOOL); Output slice = ops::Slice(root.WithOpName("slice"), input, begin, size);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
Output arg0 = ops::_Arg(s.WithOpName("arg0"), DT_INT32, 0); Output arg1 = ops::_Arg(s.WithOpName("arg1"), DT_FLOAT, 1); Output arg2 = ops::_Arg(s.WithOpName("arg2"), DT_INT32, 2); Output add = ops::Add(s.WithOpName("add"), arg0, arg0); auto ret0 = ops::_Retval(s.WithOpName("ret0"), add, 0); auto ret1 = ops::_Retval(s.WithOpName("ret1"), arg1, 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc
} DCHECK_EQ(slice_inputs.size_as_vector[i], -1); Output begin_i = ops::Slice( host_scope.WithOpName("begin_", i), slice_inputs.begin, constant_pool.Get1DHostConstant(i), constant_pool.Get1DHostConstant(1)); Output input_shape_i = ops::Slice( host_scope.WithOpName("input_shape_", i), input_shape, constant_pool.Get1DHostConstant(i), constant_pool.Get1DHostConstant(1));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 13.8K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass.cc
// outputs = Merge(tf_call_outputs, xla_run_outputs). ops::Switch s(root.WithOpName("predicated_compilation_key"), xla_compile.key, xla_compile.compilation_successful); Output predicated_compilation_key = s.output_true; Output inverse_predicated_compilation_key = s.output_false; ops::_XlaRun xla_run(root.WithOpName("xla_run"), xla_run_args,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util_test.cc
Scope root = Scope::NewRootScope().ExitOnError(); Output x = ops::Placeholder(root.WithOpName("x"), DT_FLOAT); NameAttrList f_name_attr; f_name_attr.set_name("XTimesTwoFloat"); ops::TPUPartitionedCall f(root.WithOpName("f"), {x}, /*device_ordinal=*/0, {DT_FLOAT}, f_name_attr); TF_ASSERT_OK(root.ToGraph(&graph));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 19:51:50 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
} Node* MakeWrite(const Scope& scope, Output value_to_write, const string& id) { Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT, TensorShape({})); ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id), var_handle, value_to_write); return assign_op.operation.node(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0)