Searched refs:CheckKeyShape (Results 1 – 3 of 3) sorted by relevance
24 Status LookupInterface::CheckKeyShape(const TensorShape& shape) { in CheckKeyShape() function in tensorflow::lookup::LookupInterface49 TF_RETURN_IF_ERROR(CheckKeyShape(keys.shape())); in CheckKeyAndValueTensorsHelper()79 return CheckKeyShape(keys.shape()); in CheckKeyTensorForRemove()85 TF_RETURN_IF_ERROR(CheckKeyShape(key.shape())); in CheckFindArguments()
152 Status CheckKeyShape(const TensorShape& shape);
530 TF_RETURN_IF_ERROR(CheckKeyShape(keys.shape())); in CheckKeyAndValueTensorsForImport()