- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 65 for num_inputs (0.12 sec)
-
tensorflow/cc/gradients/array_grad.cc
const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) { if (op.num_inputs() < 3) { return errors::InvalidArgument("SplitV requires 3 arguments"); } grad_outputs->push_back(Concat(scope, grad_inputs, op.input(2))); for (int i = 0; i < op.num_inputs() - 1; ++i) { grad_outputs->push_back(NoGradient()); } return scope.status(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/c/ops.cc
return cc_ctx->num_inputs(); } void TF_ShapeInferenceContextGetInput(TF_ShapeInferenceContext* ctx, int i, TF_ShapeHandle* handle, TF_Status* status) { TF_SetStatus(status, TF_OK, ""); auto* cc_ctx = reinterpret_cast<InferenceContext*>(ctx); if (i < 0 || i >= cc_ctx->num_inputs()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 28 22:41:35 UTC 2022 - 10.9K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/public/signature_def_function.h
// construction fails, `status` will be non-OK and the returned pointer will be // null. TF_CAPI_EXPORT extern TFE_Op* TF_SignatureDefFunctionMakeCallOp( TF_SignatureDefFunction* func, TFE_TensorHandle** inputs, int num_inputs, TF_Status* status); #ifdef __cplusplus } // end extern "C" #endif // __cplusplus
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 17 23:03:48 UTC 2020 - 2.1K bytes - Viewed (0) -
tensorflow/cc/framework/while_gradients_test.cc
namespace tensorflow { namespace { class WhileGradientsTest : public ::testing::Test { protected: WhileGradientsTest() : scope_(Scope::NewRootScope()) {} void Init(int num_inputs, DataType dtype = DT_INT32) { for (int i = 0; i < num_inputs; ++i) { inputs_.push_back(ops::Placeholder(scope_, dtype)); } } void CreateLoop(const ops::CondGraphBuilderFn& cond,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 27 20:32:17 UTC 2017 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
std::vector<const Edge*> in_edges; TF_RETURN_IF_ERROR(launch->input_edges(&in_edges)); const int num_inputs = in_edges.size(); const int variable_start_index = xla_function_info.variable_start_index; const int num_variables = num_inputs - variable_start_index; const int num_args = variable_start_index; VLOG(4) << "Launch node '" << launch->name() << "'"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.cc
TFE_OpAddAttrs(op, attributes); TFE_OpSetDevice(op, dev->underlying_device.c_str(), s); if (TF_GetCode(s) != TF_OK) return; int num_inputs = TFE_OpGetFlatInputCount(original_op, s); if (TF_GetCode(s) != TF_OK) return; for (int j = 0; j < num_inputs; ++j) { TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, j, s); if (TF_GetCode(s) != TF_OK) return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 03 20:47:31 UTC 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/tf_concrete_function_loading_test.cc
: TensorHandleConvertible( testing::CreateTensorHandle(ctx, DT_FLOAT, {2, 4}, value)) {} }; FunctionDef FuncDefWithNumInputsOutputs(int num_inputs, int num_outputs) { FunctionDef func; OpDef* signature = func.mutable_signature(); for (int i = 0; i < num_inputs; ++i) { signature->add_input_arg(); } for (int i = 0; i < num_outputs; ++i) { signature->add_output_arg(); } return func; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 19:16:58 UTC 2023 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
return variable_lookup; } } // anonymous namespace std::vector<const Tensor*> InputsFromContext(OpKernelContext* ctx) { std::vector<const Tensor*> inputs; inputs.reserve(ctx->num_inputs()); for (int input_idx = 0; input_idx < ctx->num_inputs(); input_idx++) { inputs.push_back(&ctx->input(input_idx)); } return inputs; } absl::StatusOr<std::vector<int>> GetConstantInputIndicesFromContext(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
absl::StatusMessageAsCStr(status)); } void TF_ExecuteOperation(TF_AbstractOp* op, int num_inputs, TF_AbstractTensor* const* inputs, TF_OutputList* o, TF_Status* s) { for (int i = 0; i < num_inputs; i++) { tsl::Set_TF_Status_from_Status(s, unwrap(op)->AddInput(unwrap(inputs[i]))); if (TF_GetCode(s) != TF_OK) { return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0) -
tensorflow/c/ops_test.cc
} #define C_CTX(x) reinterpret_cast<TF_ShapeInferenceContext*>(x) #define C_SHP(x) reinterpret_cast<TF_ShapeHandle*>(x) static OpDef MakeOpDef(int num_inputs, int num_outputs) { OpRegistrationData op_reg_data; OpDefBuilder b("dummy"); for (int i = 0; i < num_inputs; ++i) { b.Input(strings::StrCat("i", i, ": float")); } for (int i = 0; i < num_outputs; ++i) { b.Output(strings::StrCat("o", i, ": float"));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 12.6K bytes - Viewed (0)