- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for CHECK_LT (0.09 sec)
-
tensorflow/c/kernels.cc
} TF_DataType TF_InputDatatype(TF_OpKernelContext* ctx, int index) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelContext*>(ctx); CHECK_GE(index, 0); // Crash OK CHECK_LT(index, cc_ctx->num_inputs()); // Crash OK return static_cast<TF_DataType>(cc_ctx->input_dtype(index)); } void TF_SetOutput(TF_OpKernelContext* ctx, int i, const TF_Tensor* tensor,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
const XlaCompiler::ResourceUpdate& write = compilation_result->resource_updates[i]; int actual_input_index = write.input_index - missing_ctx_input_prefix; CHECK_GE(actual_input_index, 0); CHECK_LT(actual_input_index, ctx->num_inputs()); Var* var = variable_infos[variable_info_lookup[actual_input_index]].var(); CHECK(var); VLOG(2) << "Updating variable #" << i
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/xla_launch_util_test.cc
tensors_.push_back(device_tensor); return device_tensor; } // Gets the `output_index`-th output set in the context_ Tensor* GetOutput(int output_index) { CHECK_LT(output_index, context_->num_outputs()); Tensor* device_tensor = context_->mutable_output(output_index); managed_outputs_.resize(context_->num_outputs()); if (managed_outputs_[output_index]) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 28.8K bytes - Viewed (0)