Home
last modified time | relevance | path

Searched refs:GetTensorListShapeFromElementShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.h76 Status GetTensorListShapeFromElementShape(const xla::Shape& element_shape,
Dtensor_list_ops.cc172 OP_REQUIRES_OK(ctx, GetTensorListShapeFromElementShape( in Compile()
244 OP_REQUIRES_OK(ctx, GetTensorListShapeFromElementShape( in Compile()
Dtensor_list_utils.cc247 Status GetTensorListShapeFromElementShape(const xla::Shape& element_shape, in GetTensorListShapeFromElementShape() function
313 TF_RETURN_IF_ERROR(GetTensorListShapeFromElementShape( in GetInitializedTensorListForElement()