Searched refs:TensorShapeFromTensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.cc | 145 Status TensorShapeFromTensor(const Tensor& t, PartialTensorShape* out) { in TensorShapeFromTensor() function
171 TF_RETURN_IF_ERROR(TensorShapeFromTensor(c->input(index), element_shape)); in GetElementShapeFromInput()
240 OP_REQUIRES_OK(ctx, TensorShapeFromTensor(ctx->input(0), &element_shape)); in Compute()
386 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(0), &element_shape)); in Compute()
|
| D | list_kernels.h | 67 Status TensorShapeFromTensor(const Tensor& t, PartialTensorShape* out);
506 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(1), &element_shape)); in Compute()
681 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(1), &element_shape)); in Compute()
802 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(2), &element_shape)); in Compute()
|