- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for num_inputs (0.21 sec)
-
tensorflow/cc/gradients/math_grad.cc
// hence dx_k = dy for all x_k // So the gradient for AddN just transfers the incoming gradient to // all outgoing gradients. auto incoming = Identity(scope, grad_inputs[0]); for (int32_t i = 0; i < op.num_inputs(); ++i) { grad_outputs->push_back(incoming); } return scope.status(); } REGISTER_GRADIENT_OP("AddN", AddNGrad); Status PowGrad(const Scope& scope, const Operation& op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 50.7K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
.Attr("SomeDataTypeAttr: type"); static int num_inputs = 0; static int num_outputs = 0; // A kernel whose Compute function has a side-effect of updating num_inputs // and num_outputs. Various functions on TF_OpKernelContext are also // exercised. auto my_compute_func = [](void* kernel, TF_OpKernelContext* ctx) { num_inputs = TF_NumInputs(ctx); num_outputs = TF_NumOutputs(ctx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
FunctionType func_type = func.getFunctionType(); int num_inputs = func_type.getNumInputs(); int num_results = func_type.getNumResults(); // For each argument type in function's arguments, change it to uranked // tensor type if it's a variant type. SmallVector<Type, 8> updated_argument_types; updated_argument_types.reserve(num_inputs); UpdateTensorListTypes<mlir::OperandRange>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_OpAddAttrs(ret, TFE_OpGetAttrs(other)); int num_inputs = TFE_OpGetFlatInputCount(other, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); for (int input_index = 0; input_index < num_inputs; ++input_index) { TFE_TensorHandle* input = TFE_OpGetFlatInput(other, input_index, status);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/c/c_api.h
// For inputs that take a list of tensors. // inputs must point to TF_Output[num_inputs]. TF_CAPI_EXPORT extern void TF_AddInputList(TF_OperationDescription* desc, const TF_Output* inputs, int num_inputs); // Call once per control input to `desc`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
def OptimizeToSoftmax : Pat< (TFL_DivOp (TFL_ExpOp:$exp $input), (TFL_SumOp:$sum $sum_input, (Arith_ConstantOp I32ElementsAttr: $axes), ConstBoolAttrTrue), TFL_AF_None), (TFL_SoftmaxOp $input, ConstF32Attr<"1.0">), [(IsSame $exp, $sum_input), (AxesIsLastDimension $axes, $sum_input), (HasTwoUse $exp), (HasOneUse $sum)]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0)