Searched refs:GetElementShapeFromInput (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 47 Status GetElementShapeFromInput(OpKernelContext* c,
84 OP_REQUIRES_OK(c, GetElementShapeFromInput(c, *tensor_list, 1, in Compute()
186 c, GetElementShapeFromInput(c, *l, 2, &partial_element_shape)); in Compute()
251 c, GetElementShapeFromInput(c, *l, 1, &partial_element_shape)); in Compute()
312 c, GetElementShapeFromInput(c, *tensor_list, 1, &element_shape)); in Compute()
566 OP_REQUIRES_OK(c, GetElementShapeFromInput(c, *tensor_list, 2, in Compute()
|
| D | list_kernels.cc | 65 Status GetElementShapeFromInput(OpKernelContext* c, in GetElementShapeFromInput() function
|