Searched refs:get_scalar_input (Results 1 – 1 of 1) sorted by relevance
| /external/federated-compute/fcp/tensorflow/ |
| D | serve_slices_op.cc | 52 tensorflow::Status get_scalar_input(tensorflow::OpKernelContext* context, in get_scalar_input() function
126 OP_REQUIRES_OK(context, get_scalar_input(context, "callback_token", in Compute()
140 OP_REQUIRES_OK(context, get_scalar_input(context, "max_key", &max_key)); in Compute()
143 OP_REQUIRES_OK(context, get_scalar_input(context, "select_fn_initialize_op", in Compute()
154 get_scalar_input(context, "select_fn_key_input_tensor_name", in Compute()
158 OP_REQUIRES_OK(context, get_scalar_input( in Compute()
164 get_scalar_input(context, "select_fn_target_tensor_name", in Compute()
|