- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CHECK_GE (0.57 sec)
-
tensorflow/compiler/jit/xla_launch_util.cc
arguments.reserve(compilation_result->xla_input_shapes.size()); for (int i = 0; i < compilation_result->xla_input_shapes.size(); ++i) { int arg_num = compilation_result->input_mapping[i]; CHECK_GE(arg_num, missing_ctx_input_prefix); const xla::Shape& device_shape = compilation_result->xla_input_shapes[i]; const xla::Shape& host_shape = xla::ShapeUtil::DeviceShapeToHostShape(device_shape);
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/kernels.cc
tensorflow::Set_TF_Status_from_Status(args->status, status); } 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)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0)