Home
last modified time | relevance | path

Searched refs:TensorShapeFromTensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dlist_kernels.cc44 Status TensorShapeFromTensor(const Tensor& t, PartialTensorShape* out) { in TensorShapeFromTensor() function
73 TF_RETURN_IF_ERROR(TensorShapeFromTensor(c->input(index), element_shape)); in GetElementShapeFromInput()
155 OP_REQUIRES_OK(ctx, TensorShapeFromTensor(ctx->input(0), &element_shape)); in Compute()
326 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(0), &element_shape)); in Compute()
H A Dlist_kernels.h59 Status TensorShapeFromTensor(const Tensor& t, PartialTensorShape* out);
591 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(1), &element_shape)); in Compute()
778 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(1), &element_shape)); in Compute()
909 OP_REQUIRES_OK(c, TensorShapeFromTensor(c->input(2), &element_shape)); in Compute()
H A Dragged_tensor_to_tensor_op.cc157 TF_RETURN_IF_ERROR(TensorShapeFromTensor(c->input(kShapeInputIndex), in CalculateOutputSize()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc115 RuntimeShape TensorShapeFromTensor(const TfLiteTensor& tensor) { in TensorShapeFromTensor() function
261 RuntimeShape output_shape_shape = TensorShapeFromTensor(output_shape); in CalculateOutputSize()