MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetInputList

Function GetInputList

tensorflow/core/kernels/list_kernels.cc:177–190  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

175}
176
177Status GetInputList(OpKernelContext* c, int index, const TensorList** list) {
178 if (!TensorShapeUtils::IsScalar(c->input(index).shape())) {
179 return errors::InvalidArgument("Input list must be a scalar saw: ",
180 c->input(index).shape().DebugString());
181 }
182 const TensorList* l = c->input(index).scalar<Variant>()().get<TensorList>();
183 if (l == nullptr) {
184 return errors::InvalidArgument(
185 "Input handle is not a list. Saw: '",
186 c->input(index).scalar<Variant>()().DebugString(), "'");
187 }
188 *list = l;
189 return Status::OK();
190}
191
192Status ForwardInputOrCreateNewList(OpKernelContext* c, int32 input_index,
193 int32 output_index,

Callers 11

ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85
ComputeMethod · 0.85

Calls 5

InvalidArgumentFunction · 0.85
IsScalarFunction · 0.50
shapeMethod · 0.45
inputMethod · 0.45
DebugStringMethod · 0.45

Tested by

no test coverage detected