- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 23 for output_index (0.18 sec)
-
tensorflow/c/kernels.cc
return nullptr; } return tf_tensor; } TF_Tensor* TF_ForwardInputOrAllocateOutput( TF_OpKernelContext* context, const int* candidate_input_indices, int num_candidate_input_indices, int output_index, const int64_t* output_dims, int output_num_dims, int* forwarded_input, TF_Status* status) { TF_SetStatus(status, TF_OK, ""); auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/c/kernels_experimental.cc
int32_t output_index) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); if (cc_ctx->input_dtype(input_index) != tensorflow::DT_RESOURCE) { cc_ctx->forward_ref_input_to_ref_output(input_index, output_index); } } void TF_ReleaseVariableInputLockHolder(TF_VariableInputLockHolder* lockHolder) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 06:12:29 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/c/kernels.h
size_t len, TF_Status* status); // Tries to forward one of the inputs given in input_indices to // output[output_index]. If none of the given inputs can be forwarded, calls // allocate_output() to allocate a new output buffer. The index of the // forwarded input will be assign to output argument forwarded_input (if it's
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 22:46:22 UTC 2024 - 24.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
xla::ShapeIndex output_index = input_output_alias.shape().IsTuple() ? xla::ShapeIndex({output_num}) : xla::ShapeIndex({}); CHECK(input_output_alias.shape().IsTuple() || output_num == 0); if (std::optional<xla::HloInputOutputAliasConfig::Alias> alias = input_output_alias.GetAliasedParameter(output_index)) {
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/compiler/jit/deadness_analysis_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 06:59:07 UTC 2024 - 51.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor_ops.td
dead, then all of the outputs are marked as dead as well. }]; let description = [{ This is defined in TensorFlow as: REGISTER_OP("_SwitchN") .Input("data: T") .Input("output_index: int32") .Output("outputs: num_outs * T") .Attr("num_outs: int >= 1") .Attr("T: type") .SetShapeFn(SwitchNShape); For example:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 23 19:35:12 UTC 2023 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
for (const auto& entry : resource_arg_to_new_output) { int64_t resource_arg_index = entry.getFirst(); int64_t output_index = entry.getSecond(); new_retvals[output_index] = branch.getArgument(resource_arg_index); } auto old_return = branch.front().getTerminator(); OpBuilder builder(old_return); auto new_return =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
/*axis=*/0); for (auto index_item : llvm::zip(index_attr, items.getResults())) { int64_t output_index = std::get<0>(index_item).getSExtValue(); Value item = std::get<1>(index_item); values[output_index] = item; } } rewriter.replaceOpWithNewOp<PackOp>(op, op.getType(), values); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
if (num_outs.getInt() <= 0) return parser.emitError(parser.getNameLoc()) << " expects a positive number of outputs"; // `types` already contains the type for the data, add an i32 for the // output_index, and then the optional control inputs. auto builder = parser.getBuilder(); types.push_back(RankedTensorType::get({}, builder.getIntegerType(32))); Type control_type = ControlType::get(builder.getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/c/kernels_test.cc
int64_t output_dims[1] = {}; TF_Tensor* output = TF_ForwardInputOrAllocateOutput( /*context=*/ctx, candidate_input_indices, /*num_candidate_input_indices=*/1, /*output_index=*/0, output_dims, /*output_num_dims=*/0, &forwarded_input, /*status=*/s); EXPECT_EQ(TF_OK, TF_GetCode(s)); EXPECT_EQ(forwarded_input, 0); EXPECT_EQ(TF_FLOAT, TF_TensorType(output));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0)