Searched refs:TensorShapeFromTensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.cc | 42 Status TensorShapeFromTensor(const Tensor& t, PartialTensorShape* out) { in TensorShapeFromTensor() function
68 TF_RETURN_IF_ERROR(TensorShapeFromTensor(c->input(index), element_shape)); in GetElementShapeFromInput()
150 OP_REQUIRES_OK(ctx, TensorShapeFromTensor(ctx->input(0), &element_shape)); in Compute()
303 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(0), &element_shape)); in Compute()
|
| D | list_kernels.h | 45 Status TensorShapeFromTensor(const Tensor& t, PartialTensorShape* out);
483 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(1), &element_shape)); in Compute()
658 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(1), &element_shape)); in Compute()
779 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(2), &element_shape)); in Compute()
|
| D | ragged_tensor_to_tensor_op.cc | 157 TF_RETURN_IF_ERROR(TensorShapeFromTensor(c->input(kShapeInputIndex), in CalculateOutputSize()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite.cc | 115 RuntimeShape TensorShapeFromTensor(const TfLiteTensor& tensor) { in TensorShapeFromTensor() function
261 RuntimeShape output_shape_shape = TensorShapeFromTensor(output_shape); in CalculateOutputSize()
|