Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtensor_list_utils.h70 Status GetLeadingDimForTensorList(xla::XlaOp list, int64* leading_dim,
Dtensor_list_utils.cc205 Status GetLeadingDimForTensorList(xla::XlaOp list, int64* leading_dim, in GetLeadingDimForTensorList() function
302 TF_RETURN_IF_ERROR(GetLeadingDimForTensorList( in GetInitializedTensorListForElement()
Dtensor_list_ops.cc95 OP_REQUIRES_OK(ctx, GetLeadingDimForTensorList(ctx->Input(0), &leading_dim, in Compile()