Home
last modified time | relevance | path

Searched refs:kFixedInputs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_v2_ops.cc44 const int kFixedInputs = 3; // Prefix, tensor names, shape_and_slices. in ValidateInputs() local
65 FastBoundsCheck(tensor_names.NumElements() + kFixedInputs, in ValidateInputs()
74 OP_REQUIRES(context, context->num_inputs() == num_tensors + kFixedInputs, in ValidateInputs()
77 context->num_inputs() - kFixedInputs, " tensors.")); in ValidateInputs()
78 OP_REQUIRES(context, context->num_inputs() == num_tensors + kFixedInputs, in ValidateInputs()
80 "Expected a total of ", num_tensors + kFixedInputs, in ValidateInputs()
102 const int kFixedInputs = 3; // Prefix, tensor names, shape_and_slices. in Compute() local
114 const Tensor& tensor = context->input(i + kFixedInputs); in Compute()
Dsave_restore_tensor.cc55 const int kFixedInputs = save_slices ? 3 : 2; in SaveTensors() local
58 FastBoundsCheck(tensor_names_t.NumElements() + kFixedInputs, in SaveTensors()
75 OP_REQUIRES(context, context->num_inputs() == N + kFixedInputs, in SaveTensors()
76 errors::InvalidArgument("Expected totally ", N + kFixedInputs, in SaveTensors()
102 const Tensor& input = context->input(i + kFixedInputs); in SaveTensors()