Searched refs:expected_value_shape (Results 1 – 2 of 2) sorted by relevance
51 TensorShape expected_value_shape = keys.shape(); in CheckKeyAndValueTensorsHelper() local53 expected_value_shape.RemoveDim(expected_value_shape.dims() - 1); in CheckKeyAndValueTensorsHelper()55 expected_value_shape.AppendShape(value_shape()); in CheckKeyAndValueTensorsHelper()56 if (values.shape() != expected_value_shape) { in CheckKeyAndValueTensorsHelper()58 "Expected shape ", expected_value_shape.DebugString(), in CheckKeyAndValueTensorsHelper()
541 TensorShape expected_value_shape = keys.shape(); in CheckKeyAndValueTensorsForImport() local542 expected_value_shape.RemoveLastDims(key_shape.dims()); in CheckKeyAndValueTensorsForImport()543 expected_value_shape.AppendShape(value_shape); in CheckKeyAndValueTensorsForImport()544 if (values.shape() != expected_value_shape) { in CheckKeyAndValueTensorsForImport()546 "Expected shape ", expected_value_shape.DebugString(), in CheckKeyAndValueTensorsForImport()